Upload modutils.py
Browse files- modutils.py +5 -8
modutils.py
CHANGED
|
@@ -295,32 +295,29 @@ def set_lora_prompt(prompt_gui, prompt_syntax_gui, lora1_gui, lora_scale_1_gui,
|
|
| 295 |
import os
|
| 296 |
if not "Classic" in str(prompt_syntax_gui): return prompt_gui
|
| 297 |
loras = []
|
| 298 |
-
if lora1_gui:
|
| 299 |
basename = os.path.splitext(os.path.basename(lora1_gui))[0]
|
| 300 |
loras.append(f"<lora:{basename}:{lora_scale_1_gui:.2f}>")
|
| 301 |
-
if lora2_gui:
|
| 302 |
basename = os.path.splitext(os.path.basename(lora2_gui))[0]
|
| 303 |
loras.append(f"<lora:{basename}:{lora_scale_2_gui:.2f}>")
|
| 304 |
-
if lora3_gui:
|
| 305 |
basename = os.path.splitext(os.path.basename(lora3_gui))[0]
|
| 306 |
loras.append(f"<lora:{basename}:{lora_scale_3_gui:.2f}>")
|
| 307 |
-
if lora4_gui:
|
| 308 |
basename = os.path.splitext(os.path.basename(lora4_gui))[0]
|
| 309 |
loras.append(f"<lora:{basename}:{lora_scale_4_gui:.2f}>")
|
| 310 |
-
if lora5_gui:
|
| 311 |
basename = os.path.splitext(os.path.basename(lora5_gui))[0]
|
| 312 |
loras.append(f"<lora:{basename}:{lora_scale_5_gui:.2f}>")
|
| 313 |
-
|
| 314 |
tags = prompt_gui.split(",") if prompt_gui else []
|
| 315 |
prompts = []
|
| 316 |
for tag in tags:
|
| 317 |
tag = str(tag).strip()
|
| 318 |
if tag and not "<lora" in tag:
|
| 319 |
prompts.append(tag)
|
| 320 |
-
|
| 321 |
empty = [""]
|
| 322 |
prompt = ", ".join(prompts + loras + empty)
|
| 323 |
-
|
| 324 |
return gr.update(value=prompt)
|
| 325 |
|
| 326 |
|
|
|
|
| 295 |
import os
|
| 296 |
if not "Classic" in str(prompt_syntax_gui): return prompt_gui
|
| 297 |
loras = []
|
| 298 |
+
if lora1_gui and lora1_gui != "None":
|
| 299 |
basename = os.path.splitext(os.path.basename(lora1_gui))[0]
|
| 300 |
loras.append(f"<lora:{basename}:{lora_scale_1_gui:.2f}>")
|
| 301 |
+
if lora2_gui and lora2_gui != "None":
|
| 302 |
basename = os.path.splitext(os.path.basename(lora2_gui))[0]
|
| 303 |
loras.append(f"<lora:{basename}:{lora_scale_2_gui:.2f}>")
|
| 304 |
+
if lora3_gui and lora3_gui != "None":
|
| 305 |
basename = os.path.splitext(os.path.basename(lora3_gui))[0]
|
| 306 |
loras.append(f"<lora:{basename}:{lora_scale_3_gui:.2f}>")
|
| 307 |
+
if lora4_gui and lora4_gui != "None":
|
| 308 |
basename = os.path.splitext(os.path.basename(lora4_gui))[0]
|
| 309 |
loras.append(f"<lora:{basename}:{lora_scale_4_gui:.2f}>")
|
| 310 |
+
if lora5_gui and lora5_gui != "None":
|
| 311 |
basename = os.path.splitext(os.path.basename(lora5_gui))[0]
|
| 312 |
loras.append(f"<lora:{basename}:{lora_scale_5_gui:.2f}>")
|
|
|
|
| 313 |
tags = prompt_gui.split(",") if prompt_gui else []
|
| 314 |
prompts = []
|
| 315 |
for tag in tags:
|
| 316 |
tag = str(tag).strip()
|
| 317 |
if tag and not "<lora" in tag:
|
| 318 |
prompts.append(tag)
|
|
|
|
| 319 |
empty = [""]
|
| 320 |
prompt = ", ".join(prompts + loras + empty)
|
|
|
|
| 321 |
return gr.update(value=prompt)
|
| 322 |
|
| 323 |
|