Update app.py

app.py

@@ -22,14 +22,14 @@ model_dir = snapshot_download( # https://www.modelscope.cn/models/AI-ModelScope/
     ignore_file_pattern = ['onnx*'],
 )
 
-# from huggingface_hub import snapshot_download, hf_hub_download
-# snapshot_download( # 下载整个仓库; 下briaai/RMBG-2.0需要token
-#     repo_id="alibaba-pai/Wan2.1-Fun-1.3B-Control",
-#     local_dir="ckpt/Wan2.1-Fun-1.3B-Control",
-#     local_dir_use_symlinks=False,
-#     resume_download=True,
-#     repo_type="model"
-# )
+from huggingface_hub import snapshot_download, hf_hub_download
+snapshot_download( # 下载整个仓库; 下briaai/RMBG-2.0需要token
+    repo_id="alibaba-pai/Wan2.1-Fun-1.3B-Control",
+    local_dir="ckpt/Wan2.1-Fun-1.3B-Control",
+    local_dir_use_symlinks=False,
+    resume_download=True,
+    repo_type="model"
+)
 
 # hf_hub_download(
 #     repo_id="Kunbyte/Lumen",
@@ -59,12 +59,13 @@ 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:
+    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', 
+            wan_dit_path if wan_dit_path else 'ckpt/Wan2.1-Fun-1.3B-Control/diffusion_pytorch_model.safetensors', 
             'ckpt/Wan2.1-Fun-1.3B-Control/Wan2.1_VAE.pth', 
-            # 'ckpt/Wan2.1-Fun-1.3B-Control/models_t5_umt5-xxl-enc-bf16.pth', 
-            # 'ckpt/Wan2.1-Fun-1.3B-Control/models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth', 
+            'ckpt/Wan2.1-Fun-1.3B-Control/models_t5_umt5-xxl-enc-bf16.pth', 
+            'ckpt/Wan2.1-Fun-1.3B-Control/models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth', 
         ], 
         torch_dtype=torch.bfloat16, 
     )