Update app.py

app.py

@@ -38,8 +38,7 @@ snapshot_download( # 下载整个仓库
 
 model_manager = ModelManager(device="cpu") # 1.3b: device=cpu: uses 6G VRAM, device=device: uses 16G VRAM; about 1-2 min per video
 
-# 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'
-wan_dit_path = None
+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'
 if 'wan14b' in wan_dit_path.lower(): # 14B: uses about 36G, about 10 min per video 
     model_manager.load_models(
         [
@@ -50,7 +49,8 @@ if 'wan14b' in wan_dit_path.lower(): # 14B: uses about 36G, about 10 min per vid
         ], 
         torch_dtype=torch.bfloat16, # float8_e4m3fn fp8量化; bfloat16 
     ) 
-else: 
+else:
+    wan_dit_path = None
     model_manager.load_models( 
         [ 
             wan_dit_path if wan_dit_path else 'ckpt/Wan2.1-Fun-1.3B-Control/diffusion_pytorch_model.safetensors',