TheOneHong commited on
Commit
b752e9b
1 Parent(s): bffc704

fix for loras without trigger word

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -93,7 +93,7 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
93
  raise gr.Error("You must select a LoRA before proceeding.")
94
  selected_lora = loras[selected_index]
95
  lora_path = selected_lora["repo"]
96
- trigger_word = selected_lora["trigger_word"]
97
  if(trigger_word):
98
  if "trigger_position" in selected_lora:
99
  if selected_lora["trigger_position"] == "prepend":
 
93
  raise gr.Error("You must select a LoRA before proceeding.")
94
  selected_lora = loras[selected_index]
95
  lora_path = selected_lora["repo"]
96
+ trigger_word = selected_lora.get("trigger_word")
97
  if(trigger_word):
98
  if "trigger_position" in selected_lora:
99
  if selected_lora["trigger_position"] == "prepend":