optimize: sd

2024-04-02 18:21:08 +08:00 · 2024-04-02 18:21:08 +08:00 · 1c004e0df6
parent 59d279fbe0
commit 1c004e0df6
1 changed files with 1 additions and 1 deletions
--- a/api/core/tools/provider/builtin/stablediffusion/tools/stable_diffusion.py
+++ b/api/core/tools/provider/builtin/stablediffusion/tools/stable_diffusion.py
@ -189,7 +189,7 @@ class StableDiffusionTool(BuiltinTool):
            if not base_url:
                return []
            api_url = str(URL(base_url) / 'sdapi' / 'v1' / 'sd-models')
-            response = get(url=api_url, timeout=10)
+            response = get(url=api_url, timeout=(2, 10))
            if response.status_code != 200:
                return []
            else: