David Krajewski commited on
Commit
b267714
·
1 Parent(s): 87f002f

Changed output dir

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -88,8 +88,8 @@ def get_sparseflow_and_mask_forward(
88
 
89
  def download_models(ckpts_path):
90
  try:
91
- snapshot_download(repo_id="vdo/stable-video-diffusion-img2vid-xt-1-1", local_dir=ckpts_path, cache_dir=ckpts_path)
92
- snapshot_download(repo_id="MyNiuuu/MOFA-Video-Traj", local_dir=ckpts_path, cache_dir=ckpts_path, allow_patterns=["ckpts/controlnet/*"])
93
  except (Exception, BaseException) as error:
94
  print(error)
95
 
 
88
 
89
  def download_models(ckpts_path):
90
  try:
91
+ snapshot_download(repo_id="vdo/stable-video-diffusion-img2vid-xt-1-1", local_dir=f"{ckpts_path}/stable-video-diffusion-img2vid-xt-1-1", cache_dir=f"{ckpts_path}/stable-video-diffusion-img2vid-xt-1-1")
92
+ snapshot_download(repo_id="MyNiuuu/MOFA-Video-Traj", local_dir=f"{ckpts_path}/controlnet", cache_dir=f"{ckpts_path}/controlnet", allow_patterns=["ckpts/controlnet/*"])
93
  except (Exception, BaseException) as error:
94
  print(error)
95