Update app.py

app.py

@@ -53,9 +53,8 @@ def download_model_component(repo_id, subpath):
         repo_id=repo_id,
         repo_type="model",
         #local_dir=f"ckpt/{repo_id.replace('/', '--')}",
-        local_dir=f"ckpt/ROSE",
+        local_dir="ckpt/ROSE",
         local_dir_use_symlinks=False,
-        allow_patterns=[f"{subpath}/*"]
     ), subpath)
 
 # pretrained_model_path = "./models/Diffusion_Transformer/Wan2.1-Fun-1.3B-InP"
@@ -75,7 +74,7 @@ vae_path = download_component_subfolder(repo_id, config['vae_kwargs'].get('vae_s
 # 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'))
+transformer_path = download_component_subfolder("Kunbyte/ROSE", config['transformer_additional_kwargs'].get('transformer_subpath', 'transformer'))
 
 tokenizer= AutoTokenizer.from_pretrained(tokenizer_path)