Skip to content

D
dify
Unverified Commit dbcf03ef authored by Yeuoly's avatar Yeuoly
Browse files
Options

fix: change stablediffusion get models timeout

parent 2f867266
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
api/core/tools/provider/builtin/stablediffusion/tools/stable_diffusion.py
View file @ dbcf03ef
...@@ -152,7 +152,7 @@ class StableDiffusionTool(BuiltinTool): ...@@ -152,7 +152,7 @@ class StableDiffusionTool(BuiltinTool):
if not model: if not model:
raise ToolProviderCredentialValidationError('Please input model') raise ToolProviderCredentialValidationError('Please input model')
response = get(url=f'{base_url}/sdapi/v1/sd-models', timeout=120) response = get(url=f'{base_url}/sdapi/v1/sd-models', timeout=10)
if response.status_code != 200: if response.status_code != 200:
raise ToolProviderCredentialValidationError('Failed to get models') raise ToolProviderCredentialValidationError('Failed to get models')
else: else:
...@@ -172,7 +172,7 @@ class StableDiffusionTool(BuiltinTool): ...@@ -172,7 +172,7 @@ class StableDiffusionTool(BuiltinTool):
base_url = self.runtime.credentials.get('base_url', None) base_url = self.runtime.credentials.get('base_url', None)
if not base_url: if not base_url:
return [] return []
response = get(url=f'{base_url}/sdapi/v1/sd-models', timeout=120) response = get(url=f'{base_url}/sdapi/v1/sd-models', timeout=10)
if response.status_code != 200: if response.status_code != 200:
return [] return []
else: else:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment