Update app.py

app.py

@@ -38,15 +38,15 @@ def filter_kwargs(cls, kwargs):
 
 from huggingface_hub import snapshot_download
 
-# def download_component_subfolder(repo_id, subfolder):
-#     local_dir = snapshot_download(
-#         repo_id=repo_id,
-#         repo_type="model",
-#         local_dir="ckpt/Wan2.1-Fun-1.3B-InP",
-#         local_dir_use_symlinks=False,
-#         # allow_patterns=[f"{subfolder}/*"]
-#     )
-#     return os.path.join(local_dir, subfolder)
+def download_component_subfolder(repo_id, subfolder):
+    local_dir = snapshot_download(
+        repo_id=repo_id,
+        repo_type="model",
+        local_dir="ckpt/Wan2.1-Fun-1.3B-InP",
+        local_dir_use_symlinks=False,
+        # allow_patterns=[f"{subfolder}/*"]
+    )
+    return os.path.join(local_dir, subfolder)
 
 def download_model_component(repo_id, subpath):
     return os.path.join(snapshot_download(
@@ -64,19 +64,19 @@ transformer_path = "Kunbyte/ROSE"
 config_path = "configs/wan2.1/wan_civitai.yaml"
 config = OmegaConf.load(config_path)
 
-# repo_id = "alibaba-pai/Wan2.1-Fun-1.3B-InP"
+repo_id = "alibaba-pai/Wan2.1-Fun-1.3B-InP"
 
-# text_encoder_path = download_component_subfolder(repo_id, config['text_encoder_kwargs'].get('text_encoder_subpath', 'text_encoder'))
-# tokenizer_path = download_component_subfolder(repo_id, config['text_encoder_kwargs'].get('tokenizer_subpath', 'tokenizer'))
-# image_encoder_path = download_component_subfolder(repo_id, config['image_encoder_kwargs'].get('image_encoder_subpath', 'image_encoder'))
-# vae_path = download_component_subfolder(repo_id, config['vae_kwargs'].get('vae_subpath', 'vae'))
-tokenizer_path = download_model_component("alibaba-pai/Wan2.1-Fun-1.3B-InP", config['text_encoder_kwargs'].get('tokenizer_subpath', 'tokenizer'))
-text_encoder_path = download_model_component("alibaba-pai/Wan2.1-Fun-1.3B-InP", config['text_encoder_kwargs'].get('text_encoder_subpath', 'text_encoder'))
-image_encoder_path = download_model_component("alibaba-pai/Wan2.1-Fun-1.3B-InP", config['image_encoder_kwargs'].get('image_encoder_subpath', 'image_encoder'))
-vae_path = download_model_component("alibaba-pai/Wan2.1-Fun-1.3B-InP", config['vae_kwargs'].get('vae_subpath', 'vae'))
+text_encoder_path = download_component_subfolder(repo_id, config['text_encoder_kwargs'].get('text_encoder_subpath', 'text_encoder'))
+tokenizer_path = download_component_subfolder(repo_id, config['text_encoder_kwargs'].get('tokenizer_subpath', 'tokenizer'))
+image_encoder_path = download_component_subfolder(repo_id, config['image_encoder_kwargs'].get('image_encoder_subpath', 'image_encoder'))
+vae_path = download_component_subfolder(repo_id, config['vae_kwargs'].get('vae_subpath', 'vae'))
+# tokenizer_path = download_model_component("alibaba-pai/Wan2.1-Fun-1.3B-InP", config['text_encoder_kwargs'].get('tokenizer_subpath', 'tokenizer'))
+# text_encoder_path = download_model_component("alibaba-pai/Wan2.1-Fun-1.3B-InP", config['text_encoder_kwargs'].get('text_encoder_subpath', 'text_encoder'))
+# image_encoder_path = download_model_component("alibaba-pai/Wan2.1-Fun-1.3B-InP", config['image_encoder_kwargs'].get('image_encoder_subpath', 'image_encoder'))
+# vae_path = download_model_component("alibaba-pai/Wan2.1-Fun-1.3B-InP", config['vae_kwargs'].get('vae_subpath', 'vae'))
 transformer_path = download_model_component("Kunbyte/ROSE", config['transformer_additional_kwargs'].get('transformer_subpath', 'transformer'))
 
-tokenizer= AutoTokenizer.from_pretrained(tokenizer_path)
+tokenizer = AutoTokenizer.from_pretrained(tokenizer_path)
 
 text_encoder = WanT5EncoderModel.from_pretrained(
     text_encoder_path,