Upload convert_url_to_diffusers_flux_gr.py

convert_url_to_diffusers_flux_gr.py

@@ -627,10 +627,10 @@ def download_repo(repo_name, path, use_original=["vae", "text_encoder"], progres
     from huggingface_hub import snapshot_download
     print(f"Downloading {repo_name}.")
     try:
-        if "text_encoder_2" not in use_original:
-            snapshot_download(repo_id=repo_name, local_dir=path, ignore_patterns=["transformer/", "*.sft", ".*", "README*", "*.md", "*.index", "*.jpg", "*.png", "*.webp"])
+        if "text_encoder_2" in use_original:
+            snapshot_download(repo_id=repo_name, local_dir=path, ignore_patterns=["transformer/diffusion*.*", "*.sft", ".*", "README*", "*.md", "*.index", "*.jpg", "*.png", "*.webp"])
         else:
-            snapshot_download(repo_id=repo_name, local_dir=path, ignore_patterns=["transformer/", "text_encoder_2/", "*.sft", ".*", "README*", "*.md", "*.index", "*.jpg", "*.png", "*.webp"])
+            snapshot_download(repo_id=repo_name, local_dir=path, ignore_patterns=["transformer/diffusion*.*", "text_encoder_2/model*.*", "*.sft", ".*", "README*", "*.md", "*.index", "*.jpg", "*.png", "*.webp"])
     except Exception as e:
         print(e)
 
@@ -656,7 +656,7 @@ def copy_nontensor_files(from_path, to_path, use_original=["vae", "text_encoder"
     print(f"Copying Tokenizer 2 files {tn2_from} to {tn2_to}")
     shutil.copytree(tn2_from, tn2_to, ignore=shutil.ignore_patterns(".*", "README*", "*.md", "*.jpg", "*.png", "*.webp"), dirs_exist_ok=True)
     print(f"Copying non-tensor files {from_path} to {to_path}")
-    shutil.copytree(from_path, to_path, ignore=shutil.ignore_patterns("*.safetensors", "*.bin", "*.sft", ".*", "README*", "*.md", "*.index", "*.jpg", "*.png", "*.webp"), dirs_exist_ok=True)
+    shutil.copytree(from_path, to_path, ignore=shutil.ignore_patterns("*.safetensors", "*.bin", "*.sft", ".*", "README*", "*.md", "*.index", "*.jpg", "*.png", "*.webp", "*.index.json"), dirs_exist_ok=True)
     
 def save_flux_other_diffusers(path: str, model_type: str = "dev", use_original: list = ["vae", "text_encoder"], progress=gr.Progress(track_tqdm=True)):
     import shutil