Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ snapshot_download( # 下载整个仓库; 下briaai/RMBG-2.0需要token
|
|
38 |
hf_hub_download(
|
39 |
repo_id="Kunbyte/Lumen",
|
40 |
filename="Lumen-T2V-1.3B-V1.0.ckpt",
|
41 |
-
local_dir="ckpt/",
|
42 |
local_dir_use_symlinks=False,
|
43 |
resume_download=True,
|
44 |
)
|
@@ -49,7 +49,7 @@ rmbg_model.to(device)
|
|
49 |
rmbg_model.eval()
|
50 |
|
51 |
model_manager = ModelManager(device="cpu") # 1.3b: device=cpu: uses 6G VRAM, device=device: uses 16G VRAM; about 1-2 min per video
|
52 |
-
wan_dit_path = '
|
53 |
|
54 |
if 'wan14b' in wan_dit_path.lower(): # 14B: uses about 36G, about 10 min per video
|
55 |
model_manager.load_models(
|
|
|
38 |
hf_hub_download(
|
39 |
repo_id="Kunbyte/Lumen",
|
40 |
filename="Lumen-T2V-1.3B-V1.0.ckpt",
|
41 |
+
local_dir="ckpt/Lumen",
|
42 |
local_dir_use_symlinks=False,
|
43 |
resume_download=True,
|
44 |
)
|
|
|
49 |
rmbg_model.eval()
|
50 |
|
51 |
model_manager = ModelManager(device="cpu") # 1.3b: device=cpu: uses 6G VRAM, device=device: uses 16G VRAM; about 1-2 min per video
|
52 |
+
wan_dit_path = 'ckpt/Lumen/Lumen-T2V-1.3B-V1.0.ckpt'
|
53 |
|
54 |
if 'wan14b' in wan_dit_path.lower(): # 14B: uses about 36G, about 10 min per video
|
55 |
model_manager.load_models(
|