fast-stable-diffusion

Running

zenafey commited on Oct 27, 2023

Commit

8f278d3

•

1 Parent(s): c84504e

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -57,9 +57,9 @@ def place_lora(current_prompt, lora_name):
     pattern = r"<lora:" + lora_name + r":.*?>"
     if re.search(pattern, current_prompt):
-        yield re.sub(pattern, "", current_prompt)
     else:
-        yield current_prompt + " <lora:" + lora_name + ":1> "
 def plaintext_to_html(text, classname=None):

     pattern = r"<lora:" + lora_name + r":.*?>"
     if re.search(pattern, current_prompt):
+        return re.sub(pattern, "", current_prompt)
     else:
+        return current_prompt + " <lora:" + lora_name + ":1> "
 def plaintext_to_html(text, classname=None):