Upload 4 files

convert_url_to_diffusers_sd.py

@@ -212,6 +212,7 @@ def fuse_loras(pipe, civitai_key="", lora_dict={}, temp_dir="."):
     pipe.set_adapters(a_list, adapter_weights=w_list)
     pipe.fuse_lora(adapter_names=a_list, lora_scale=1.0)
     pipe.unload_lora_weights()
+    return pipe
 
 
 def convert_url_to_diffusers_sd(url, civitai_key="", half=True, vae=None, scheduler="Euler", lora_dict={},
@@ -259,7 +260,7 @@ def convert_url_to_diffusers_sd(url, civitai_key="", half=True, vae=None, schedu
             elif new_vae_file:
                 pipe.vae = AutoencoderKL.from_single_file(new_vae_file)
 
-    fuse_loras(pipe, lora_dict, temp_dir, civitai_key)
+    pipe = fuse_loras(pipe, lora_dict, temp_dir, civitai_key)
 
     sconf = get_scheduler_config(scheduler)
     pipe.scheduler = sconf[0].from_config(pipe.scheduler.config, **sconf[1])

convert_url_to_diffusers_sd_gr.py

@@ -241,6 +241,7 @@ def fuse_loras(pipe, lora_dict={}, temp_dir=".", civitai_key="", hf_token=""):
     pipe.set_adapters(a_list, adapter_weights=w_list)
     pipe.fuse_lora(adapter_names=a_list, lora_scale=1.0)
     pipe.unload_lora_weights()
+    return pipe
 
 
 def convert_url_to_diffusers_sd(url, civitai_key="", hf_token="", is_upload_sf=False, half=True, vae=None, scheduler="Euler", lora_dict={},
@@ -289,7 +290,7 @@ def convert_url_to_diffusers_sd(url, civitai_key="", hf_token="", is_upload_sf=F
             elif new_vae_file:
                 pipe.vae = AutoencoderKL.from_single_file(new_vae_file)
 
-    fuse_loras(pipe, lora_dict, temp_dir, civitai_key, hf_token)
+    pipe = fuse_loras(pipe, lora_dict, temp_dir, civitai_key, hf_token)
 
     sconf = get_scheduler_config(scheduler)
     pipe.scheduler = sconf[0].from_config(pipe.scheduler.config, **sconf[1])

local/convert_url_to_diffusers_sd.py

@@ -212,6 +212,7 @@ def fuse_loras(pipe, civitai_key="", lora_dict={}, temp_dir="."):
     pipe.set_adapters(a_list, adapter_weights=w_list)
     pipe.fuse_lora(adapter_names=a_list, lora_scale=1.0)
     pipe.unload_lora_weights()
+    return pipe
 
 
 def convert_url_to_diffusers_sd(url, civitai_key="", half=True, vae=None, scheduler="Euler", lora_dict={},
@@ -259,7 +260,7 @@ def convert_url_to_diffusers_sd(url, civitai_key="", half=True, vae=None, schedu
             elif new_vae_file:
                 pipe.vae = AutoencoderKL.from_single_file(new_vae_file)
 
-    fuse_loras(pipe, lora_dict, temp_dir, civitai_key)
+    pipe = fuse_loras(pipe, lora_dict, temp_dir, civitai_key)
 
     sconf = get_scheduler_config(scheduler)
     pipe.scheduler = sconf[0].from_config(pipe.scheduler.config, **sconf[1])