wzhouxiff commited on
Commit
6e6ddbe
1 Parent(s): 83719d4

update download model name

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -248,7 +248,7 @@ 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 .')
252
 
253
  config = OmegaConf.load(config_path)
254
  model_config = config.pop("model", OmegaConf.create())
@@ -261,7 +261,7 @@ 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 .')
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/motionctrl.pth')
252
 
253
  config = OmegaConf.load(config_path)
254
  model_config = config.pop("model", OmegaConf.create())
 
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)