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

Update download/download.py

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