fix: missing prompt (#2294)

2024-01-30 17:00:50 +08:00
parent 0518da5819
commit cb27571e9f
1 changed files with 2 additions and 0 deletions
--- a/api/core/tools/provider/builtin/stablediffusion/tools/stable_diffusion.py
+++ b/api/core/tools/provider/builtin/stablediffusion/tools/stable_diffusion.py
@@ -188,6 +188,8 @@ class StableDiffusionTool(BuiltinTool):

        if lora:
            draw_options['prompt'] = f'{lora},{prompt}'
+        else:
+            draw_options['prompt'] = prompt

        draw_options['width'] = width
        draw_options['height'] = height