Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,8 @@ snapshot_download( # 下载整个仓库
|
|
38 |
|
39 |
model_manager = ModelManager(device="cpu") # 1.3b: device=cpu: uses 6G VRAM, device=device: uses 16G VRAM; about 1-2 min per video
|
40 |
|
41 |
-
wan_dit_path = 'train_res/wan1.3b_zh/full_wc0.5_f1gt0.5_real1_2_zh_en_l_s/lightning_logs/version_0/checkpoints/step-step=30000.ckpt'
|
|
|
42 |
if 'wan14b' in wan_dit_path.lower(): # 14B: uses about 36G, about 10 min per video
|
43 |
model_manager.load_models(
|
44 |
[
|
|
|
38 |
|
39 |
model_manager = ModelManager(device="cpu") # 1.3b: device=cpu: uses 6G VRAM, device=device: uses 16G VRAM; about 1-2 min per video
|
40 |
|
41 |
+
# wan_dit_path = 'train_res/wan1.3b_zh/full_wc0.5_f1gt0.5_real1_2_zh_en_l_s/lightning_logs/version_0/checkpoints/step-step=30000.ckpt'
|
42 |
+
wan_dit_path = None
|
43 |
if 'wan14b' in wan_dit_path.lower(): # 14B: uses about 36G, about 10 min per video
|
44 |
model_manager.load_models(
|
45 |
[
|