wzhouxiff commited on
Commit
84e4bd0
1 Parent(s): 6e6ddbe

update download model name

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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/motionctrl.pth')
 
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/videocrafter2_motionctrl_cmcm.ckpt')
 
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)