Spaces:
Paused
Paused
Update download/download.py
Browse files- 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()
|