Upload 4 files

convert_url_to_diffusers_sdxl.py

@@ -194,7 +194,7 @@ tags:
 
 
 def fuse_loras(pipe, civitai_key="", lora_dict={}, temp_dir="."):
-    if not lora_dict or not isinstance(lora_dict, dict) or not lora_dict.keys(): return
+    if not lora_dict or not isinstance(lora_dict, dict): return
     a_list = []
     w_list = []
     for k, v in lora_dict.items():
@@ -284,7 +284,8 @@ if __name__ == "__main__":
     assert args.url is not None, "Must provide a URL!"
 
     lora_dict = {args.lora1: args.lora1s, args.lora2: args.lora2s, args.lora3: args.lora3s, args.lora4: args.lora4s, args.lora5: args.lora5s}
-    if None in lora_dict.keys(): del lora_dict[None]
+    for k, v in lora_dict.items():
+        if k == None or k == "": del lora_dict[k]
 
     if args.loras and Path(args.loras).exists():
         for p in Path(args.loras).glob('**/*.safetensors'):

convert_url_to_diffusers_sdxl_gr.py

@@ -311,7 +311,8 @@ def convert_url_to_diffusers_repo(dl_url, hf_user, hf_repo, hf_token, civitai_ke
     if hf_token and not os.environ.get("HF_TOKEN"): os.environ['HF_TOKEN'] = hf_token
     if not civitai_key and os.environ.get("CIVITAI_API_KEY"): civitai_key = os.environ.get("CIVITAI_API_KEY")
     lora_dict = {lora1: lora1s, lora2: lora2s, lora3: lora3s, lora4: lora4s, lora5: lora5s}
-    if None in lora_dict.keys(): del lora_dict[None]
+    for k, v in lora_dict.items():
+        if k == None or k == "": del lora_dict[k]
     new_path = convert_url_to_diffusers_sdxl(dl_url, civitai_key, is_upload_sf, half, vae, scheduler, lora_dict)
     if not new_path: return ""
     new_repo_id = f"{hf_user}/{Path(new_path).stem}"
@@ -357,7 +358,8 @@ if __name__ == "__main__":
     assert args.url is not None, "Must provide a URL!"
 
     lora_dict = {args.lora1: args.lora1s, args.lora2: args.lora2s, args.lora3: args.lora3s, args.lora4: args.lora4s, args.lora5: args.lora5s}
-    if None in lora_dict.keys(): del lora_dict[None]
+    for k, v in lora_dict.items():
+        if k == None or k == "": del lora_dict[k]
 
     if args.loras and Path(args.loras).exists():
         for p in Path(args.loras).glob('**/*.safetensors'):

local/convert_url_to_diffusers_sdxl.py

@@ -194,7 +194,7 @@ tags:
 
 
 def fuse_loras(pipe, civitai_key="", lora_dict={}, temp_dir="."):
-    if not lora_dict or not isinstance(lora_dict, dict) or not lora_dict.keys(): return
+    if not lora_dict or not isinstance(lora_dict, dict): return
     a_list = []
     w_list = []
     for k, v in lora_dict.items():
@@ -284,7 +284,8 @@ if __name__ == "__main__":
     assert args.url is not None, "Must provide a URL!"
 
     lora_dict = {args.lora1: args.lora1s, args.lora2: args.lora2s, args.lora3: args.lora3s, args.lora4: args.lora4s, args.lora5: args.lora5s}
-    if None in lora_dict.keys(): del lora_dict[None]
+    for k, v in lora_dict.items():
+        if k == None or k == "": del lora_dict[k]
 
     if args.loras and Path(args.loras).exists():
         for p in Path(args.loras).glob('**/*.safetensors'):