John6666 commited on
Commit
2f4e978
·
verified ·
1 Parent(s): 91e1c69

Upload modutils.py

Browse files
Files changed (1) hide show
  1. 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