Spaces:
Runtime error
Runtime error
update download model name
Browse files
app.py
CHANGED
@@ -248,7 +248,8 @@ def fn_vis_traj():
|
|
248 |
model_path='./checkpoints/motionctrl.pth'
|
249 |
config_path='./configs/inference/config_both.yaml'
|
250 |
if not os.path.exists(model_path):
|
251 |
-
os.system(f'wget https://huggingface.co/TencentARC/MotionCtrl/resolve/main/motionctrl.pth?download=true -P ./checkpoints/
|
|
|
252 |
|
253 |
config = OmegaConf.load(config_path)
|
254 |
model_config = config.pop("model", OmegaConf.create())
|
@@ -261,7 +262,8 @@ model_v1.eval()
|
|
261 |
|
262 |
v2_model_path = './checkpoints/videocrafter2_motionctrl_cmcm.ckpt'
|
263 |
if not os.path.exists(v2_model_path):
|
264 |
-
os.system(f'wget https://huggingface.co/TencentARC/MotionCtrl/resolve/main/videocrafter2_motionctrl_cmcm.ckpt?download=true -P ./checkpoints/
|
|
|
265 |
|
266 |
model_v2 = instantiate_from_config(model_config)
|
267 |
model_v2 = load_model_checkpoint(model_v2, v2_model_path)
|
|
|
248 |
model_path='./checkpoints/motionctrl.pth'
|
249 |
config_path='./configs/inference/config_both.yaml'
|
250 |
if not os.path.exists(model_path):
|
251 |
+
os.system(f'wget https://huggingface.co/TencentARC/MotionCtrl/resolve/main/motionctrl.pth?download=true -P ./checkpoints/')
|
252 |
+
os.system(f'mv ./checkpoints/motionctrl.pth?download=true ./checkpoints/motionctrl.pth')
|
253 |
|
254 |
config = OmegaConf.load(config_path)
|
255 |
model_config = config.pop("model", OmegaConf.create())
|
|
|
262 |
|
263 |
v2_model_path = './checkpoints/videocrafter2_motionctrl_cmcm.ckpt'
|
264 |
if not os.path.exists(v2_model_path):
|
265 |
+
os.system(f'wget https://huggingface.co/TencentARC/MotionCtrl/resolve/main/videocrafter2_motionctrl_cmcm.ckpt?download=true -P ./checkpoints/')
|
266 |
+
os.system(f'mv ./checkpoints/videocrafter2_motionctrl_cmcm.ckpt?download=true ./checkpoints/videocrafter2_motionctrl_cmcm.ckpt')
|
267 |
|
268 |
model_v2 = instantiate_from_config(model_config)
|
269 |
model_v2 = load_model_checkpoint(model_v2, v2_model_path)
|