Fly-ShuAI commited on
Commit
ee1ee3b
·
verified ·
1 Parent(s): c213fc7

Update app.py

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