wzhouxiff commited on
Commit
5f3a61f
1 Parent(s): 84e4bd0

update download model name

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -260,10 +260,10 @@ if torch.cuda.is_available():
260
  model_v1 = load_model_checkpoint(model_v1, model_path)
261
  model_v1.eval()
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)
 
260
  model_v1 = load_model_checkpoint(model_v1, model_path)
261
  model_v1.eval()
262
 
263
+ v2_model_path = './checkpoints/motionctrl_videocrafter2_cmcm.ckpt'
264
  if not os.path.exists(v2_model_path):
265
+ os.system(f'wget https://huggingface.co/TencentARC/MotionCtrl/resolve/main/motionctrl_videocrafter2_cmcm.ckpt?download=true -P ./checkpoints/')
266
+ os.system(f'mv ./checkpoints/motionctrl_videocrafter2_cmcm.ckpt?download=true ./checkpoints/motionctrl_videocrafter2_cmcm.ckpt')
267
 
268
  model_v2 = instantiate_from_config(model_config)
269
  model_v2 = load_model_checkpoint(model_v2, v2_model_path)