Spaces:
Running
on
Zero
Running
on
Zero
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 |
|