Browse Source

fix: missing prompt (#2294)

Yeuoly 1 year ago
parent
commit
cb27571e9f

+ 2 - 0
api/core/tools/provider/builtin/stablediffusion/tools/stable_diffusion.py

@@ -188,6 +188,8 @@ class StableDiffusionTool(BuiltinTool):
 
 
         if lora:
         if lora:
             draw_options['prompt'] = f'{lora},{prompt}'
             draw_options['prompt'] = f'{lora},{prompt}'
+        else:
+            draw_options['prompt'] = prompt
 
 
         draw_options['width'] = width
         draw_options['width'] = width
         draw_options['height'] = height
         draw_options['height'] = height