Spaces:

yanboding
/

MTVCrafter

Runtime error

yanboding commited on 20 days ago

Commit

95b36fc

verified ·

1 Parent(s): bb59364

Update inference_engine.py

Files changed (1) hide show

inference_engine.py CHANGED Viewed

@@ -16,9 +16,9 @@ def run_inference(device, motion_data_path, ref_image_path='', dst_width=512, ds
     num_frames = 49
     to_pil = ToPILImage()
     normalize = transforms.Normalize([0.5], [0.5])
-    pretrained_model_path = "/gemini/space/human_guozz2/dyb/models/CogVideoX"
-    transformer_path = "/gemini/space/human_guozz2/dyb/models/MTVCrafter/MV-DiT/CogVideoX"
-    tokenizer_path = "/gemini/space/human_guozz2/dyb/models/MTVCrafter/4DMoT/mp_rank_00_model_states.pt"
     with open(motion_data_path, 'rb') as f:
         data_list = pickle.load(f)

     num_frames = 49
     to_pil = ToPILImage()
     normalize = transforms.Normalize([0.5], [0.5])
+    pretrained_model_path = "THUDM/CogVideoX-5b"
+    transformer_path = "./MV-DiT-7B"
+    tokenizer_path = "./4DMoT/mp_rank_00_model_states.pt"
     with open(motion_data_path, 'rb') as f:
         data_list = pickle.load(f)