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

Update download/download.py

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