Update inference_engine.py

inference_engine.py

@@ -17,8 +17,8 @@ def run_inference(device, motion_data_path, ref_image_path='', dst_width=512, ds
     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"
+    transformer_path = "yanboding/MTVCrafter/MV-DiT/CogVideoX"
+    tokenizer_path = "mp_rank_00_model_states.pt"
     
     with open(motion_data_path, 'rb') as f:
         data_list = pickle.load(f)