Upload 4 files

convert_url_to_diffusers_sdxl.py

@@ -198,6 +198,7 @@ def fuse_loras(pipe, civitai_key="", lora_dict={}, temp_dir="."):
     a_list = []
     w_list = []
     for k, v in lora_dict.items():
+        if not k: continue
         new_lora_file = get_download_file(temp_dir, k, civitai_key)
         if not new_lora_file or not Path(new_lora_file).exists():
             print(f"LoRA not found: {k}")
@@ -206,7 +207,8 @@ def fuse_loras(pipe, civitai_key="", lora_dict={}, temp_dir="."):
         a_name = Path(new_lora_file).stem
         pipe.load_lora_weights(new_lora_file, weight_name = w_name, adapter_name = a_name)
         a_list.append(a_name)
-        w_list.append(v) 
+        w_list.append(v)
+    if not a_list: return
     pipe.set_adapters(a_list, adapter_weights=w_list)
     pipe.fuse_lora(adapter_names=a_list, lora_scale=1.0)
     pipe.unload_lora_weights()
@@ -284,8 +286,6 @@ 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}
-    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

@@ -199,6 +199,7 @@ def fuse_loras(pipe, lora_dict={}, temp_dir=".", civitai_key=""):
     a_list = []
     w_list = []
     for k, v in lora_dict.items():
+        if not k: continue
         new_lora_file = get_download_file(temp_dir, k, civitai_key)
         if not new_lora_file or not Path(new_lora_file).exists():
             print(f"LoRA not found: {k}")
@@ -312,8 +313,6 @@ 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}
-    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}"
@@ -359,8 +358,6 @@ 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}
-    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

@@ -198,6 +198,7 @@ def fuse_loras(pipe, civitai_key="", lora_dict={}, temp_dir="."):
     a_list = []
     w_list = []
     for k, v in lora_dict.items():
+        if not k: continue
         new_lora_file = get_download_file(temp_dir, k, civitai_key)
         if not new_lora_file or not Path(new_lora_file).exists():
             print(f"LoRA not found: {k}")
@@ -206,7 +207,8 @@ def fuse_loras(pipe, civitai_key="", lora_dict={}, temp_dir="."):
         a_name = Path(new_lora_file).stem
         pipe.load_lora_weights(new_lora_file, weight_name = w_name, adapter_name = a_name)
         a_list.append(a_name)
-        w_list.append(v) 
+        w_list.append(v)
+    if not a_list: return
     pipe.set_adapters(a_list, adapter_weights=w_list)
     pipe.fuse_lora(adapter_names=a_list, lora_scale=1.0)
     pipe.unload_lora_weights()
@@ -284,8 +286,6 @@ 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}
-    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'):