MyNiuuu
/

MOFA-Video-Hybrid

Model card Files Files and versions Community

MyNiuuu commited on Jun 22, 2024

Commit

6f8faf1

·

1 Parent(s): f5bbdc7

update run_gradio_audio_driven

Files changed (2) hide show

.DS_Store +0 -0
run_gradio_audio_driven.py +1 -1

.DS_Store CHANGED Viewed

Binary files a/.DS_Store and b/.DS_Store differ

run_gradio_audio_driven.py CHANGED Viewed

@@ -332,7 +332,7 @@ class Drag:
     def __init__(self, device, height, width, model_length):
         self.device = device
-        pretrained_model_name_or_path = "/apdcephfs/share_1290939/vg_zoo/huggingface/stable-video-diffusion-img2vid-xt-1-1"
         self.device = 'cuda'
         self.weight_dtype = torch.float16

     def __init__(self, device, height, width, model_length):
         self.device = device
+        pretrained_model_name_or_path = "./ckpts/mofa/stable-video-diffusion-img2vid-xt-1-1"
         self.device = 'cuda'
         self.weight_dtype = torch.float16