Fly-ShuAI commited on
Commit
aae145c
·
verified ·
1 Parent(s): a6581cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
  [