roll-ai commited on
Commit
2e9acb3
·
verified ·
1 Parent(s): 0c821d0

Update download/download.py

Browse files
Files changed (1) hide show
  1. download/download.py +4 -4
download/download.py CHANGED
@@ -4,22 +4,22 @@ def download_model():
4
  snapshot_download(
5
  repo_id="tencent/DepthCrafter", # 4 GB
6
  local_dir="/app/pretrained/DepthCrafter",
7
-
8
  )
9
  snapshot_download(
10
  repo_id="stabilityai/stable-video-diffusion-img2vid", # 25 gb
11
  local_dir="/app/pretrained/stable-video-diffusion-img2vid",
12
-
13
  )
14
  snapshot_download(
15
  repo_id= "Qwen/Qwen2.5-VL-7B-Instruct", # 16 GB
16
  local_dir="/app/pretrained/Qwen2.5-VL-7B-Instruct",
17
-
18
  )
19
  snapshot_download(
20
  repo_id="THUDM/CogVideoX-5b-I2V", # 22 GB
21
  local_dir="/app/pretrained/CogVideoX-5b-I2V",
22
-
23
  )
24
 
25
  download_model()
 
4
  snapshot_download(
5
  repo_id="tencent/DepthCrafter", # 4 GB
6
  local_dir="/app/pretrained/DepthCrafter",
7
+ resume_download=True,
8
  )
9
  snapshot_download(
10
  repo_id="stabilityai/stable-video-diffusion-img2vid", # 25 gb
11
  local_dir="/app/pretrained/stable-video-diffusion-img2vid",
12
+ resume_download=True,
13
  )
14
  snapshot_download(
15
  repo_id= "Qwen/Qwen2.5-VL-7B-Instruct", # 16 GB
16
  local_dir="/app/pretrained/Qwen2.5-VL-7B-Instruct",
17
+ resume_download=True,
18
  )
19
  snapshot_download(
20
  repo_id="THUDM/CogVideoX-5b-I2V", # 22 GB
21
  local_dir="/app/pretrained/CogVideoX-5b-I2V",
22
+ resume_download=True,
23
  )
24
 
25
  download_model()