bugfix: model str maybe empty (#2660)

bc65ee10 · nan jiang · GitHub · 20014836 · bc65ee10
Unverified Commit bc65ee10 authored Mar 03, 2024 by nan jiang Committed by GitHub Mar 03, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

stable_diffusion.py ...rovider/builtin/stablediffusion/tools/stable_diffusion.py +1 -1

No files found.
--- a/api/core/tools/provider/builtin/stablediffusion/tools/stable_diffusion.py
+++ b/api/core/tools/provider/builtin/stablediffusion/tools/stable_diffusion.py
@@ -70,7 +70,7 @@ class StableDiffusionTool(BuiltinTool):
        if not base_url:
            return self.create_text_message('Please input base_url')

-        if 'model' in tool_parameters:
+        if 'model' in tool_parameters and tool_parameters['model']:
            self.runtime.credentials['model'] = tool_parameters['model']

        model = self.runtime.credentials.get('model', None)