Spaces:
Running
Running
Upload convert_url_to_diffusers_sdxl_gr.py
Browse files
convert_url_to_diffusers_sdxl_gr.py
CHANGED
@@ -222,13 +222,13 @@ tags:
|
|
222 |
f.write(md)
|
223 |
|
224 |
|
225 |
-
def fuse_loras(pipe, lora_dict={}, temp_dir=".", civitai_key=""):
|
226 |
if not lora_dict or not isinstance(lora_dict, dict): return
|
227 |
a_list = []
|
228 |
w_list = []
|
229 |
for k, v in lora_dict.items():
|
230 |
if not k: continue
|
231 |
-
new_lora_file = get_download_file(temp_dir, k, civitai_key)
|
232 |
if not new_lora_file or not Path(new_lora_file).exists():
|
233 |
print(f"LoRA not found: {k}")
|
234 |
continue
|
@@ -279,7 +279,7 @@ def convert_url_to_diffusers_sdxl(url, civitai_key="", hf_token="", is_upload_sf
|
|
279 |
elif new_vae_file:
|
280 |
pipe.vae = AutoencoderKL.from_single_file(new_vae_file)
|
281 |
|
282 |
-
fuse_loras(pipe, lora_dict, temp_dir, civitai_key)
|
283 |
|
284 |
sconf = get_scheduler_config(scheduler)
|
285 |
pipe.scheduler = sconf[0].from_config(pipe.scheduler.config, **sconf[1])
|
|
|
222 |
f.write(md)
|
223 |
|
224 |
|
225 |
+
def fuse_loras(pipe, lora_dict={}, temp_dir=".", civitai_key="", hf_token=""):
|
226 |
if not lora_dict or not isinstance(lora_dict, dict): return
|
227 |
a_list = []
|
228 |
w_list = []
|
229 |
for k, v in lora_dict.items():
|
230 |
if not k: continue
|
231 |
+
new_lora_file = get_download_file(temp_dir, k, civitai_key, hf_token)
|
232 |
if not new_lora_file or not Path(new_lora_file).exists():
|
233 |
print(f"LoRA not found: {k}")
|
234 |
continue
|
|
|
279 |
elif new_vae_file:
|
280 |
pipe.vae = AutoencoderKL.from_single_file(new_vae_file)
|
281 |
|
282 |
+
fuse_loras(pipe, lora_dict, temp_dir, civitai_key, hf_token)
|
283 |
|
284 |
sconf = get_scheduler_config(scheduler)
|
285 |
pipe.scheduler = sconf[0].from_config(pipe.scheduler.config, **sconf[1])
|