Update webgui.py

webgui.py

@@ -25,8 +25,9 @@ from moviepy.editor import VideoFileClip, AudioFileClip
 from facenet_pytorch import MTCNN
 import argparse
 
-from transformers.utils import move_cache
-move_cache()
+# キャッシュの移行を明示的に実行
+import transformers
+transformers.utils.move_cache()
 
 import gradio as gr
 
@@ -122,7 +123,9 @@ denoising_unet.load_state_dict(torch.load(config.denoising_unet_path, map_locati
 ## face locator init
 # face_locator = FaceLocator(320, conditioning_channels=1, block_out_channels=(16, 32, 96, 256)).to(dtype=weight_dtype, device="cuda")
 face_locator = FaceLocator(320, conditioning_channels=1, block_out_channels=(16, 32, 96, 256)).to(dtype=weight_dtype, device="cpu")
-face_locator.load_state_dict(torch.load(config.face_locator_path))
+# face_locator.load_state_dict(torch.load(config.face_locator_path))
+face_locator.load_state_dict(torch.load(config.face_locator_path, map_location="cpu"), strict=False)
+
 
 ## load audio processor params
 audio_processor = load_audio_model(model_path=config.audio_model_path, device=device)