Spaces:

yanboding
/

MTVCrafter

Runtime error

yanboding commited on May 29

Commit

bb59364

verified ·

1 Parent(s): e0436c7

Update motion_extractor.py

Files changed (1) hide show

motion_extractor.py CHANGED Viewed

@@ -7,7 +7,7 @@ import pickle
 import torchvision
 # Load the TorchScript model once at the top
-model_path = '/gemini/space/human_guozz2/dyb/MTVCrafter-main/nlf_l_multi_0.3.2.torchscript'
 assert os.path.exists(model_path), f"Model file not found at {model_path}"
 model = torch.jit.load(model_path).cuda().eval()

 import torchvision
 # Load the TorchScript model once at the top
+model_path = 'nlf_l_multi_0.3.2.torchscript'
 assert os.path.exists(model_path), f"Model file not found at {model_path}"
 model = torch.jit.load(model_path).cuda().eval()