AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 11:15:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add stable-diffusion-3-5-large for the text-to-image tool with siliconflow (#9909)

Browse Source
This commit is contained in:
zhuhao 2024-10-27 21:17:36 +08:00 committed by GitHub
parent 8bb5b943d7
commit aa11141660
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
api/core/tools/provider/builtin/siliconflow/tools/stable_diffusion.py
View File

@ -5,9 +5,12 @@ import requests
from core.tools.entities.tool_entities import ToolInvokeMessage
from core.tools.tool.builtin_tool import BuiltinTool
SDURL = {
"sd_3": "https://api.siliconflow.cn/v1/stabilityai/stable-diffusion-3-medium/text-to-image",
"sd_xl": "https://api.siliconflow.cn/v1/stabilityai/stable-diffusion-xl-base-1.0/text-to-image",
SILICONFLOW_API_URL = "https://api.siliconflow.cn/v1/image/generations"
SD_MODELS = {
"sd_3": "stabilityai/stable-diffusion-3-medium",
"sd_xl": "stabilityai/stable-diffusion-xl-base-1.0",
"sd_3.5_large": "stabilityai/stable-diffusion-3-5-large",
}
@ -22,9 +25,10 @@ class StableDiffusionTool(BuiltinTool):
}
model = tool_parameters.get("model", "sd_3")
url = SDURL.get(model)
sd_model = SD_MODELS.get(model)
payload = {
"model": sd_model,
"prompt": tool_parameters.get("prompt"),
"negative_prompt": tool_parameters.get("negative_prompt", ""),
"image_size": tool_parameters.get("image_size", "1024x1024"),
@ -34,7 +38,7 @@ class StableDiffusionTool(BuiltinTool):
"num_inference_steps": tool_parameters.get("num_inference_steps", 20),
}
response = requests.post(url, json=payload, headers=headers)
response = requests.post(SILICONFLOW_API_URL, json=payload, headers=headers)
if response.status_code != 200:
return self.create_text_message(f"Got Error Response:{response.text}")

3
api/core/tools/provider/builtin/siliconflow/tools/stable_diffusion.yaml
View File

@ -40,6 +40,9 @@ parameters:
- value: sd_xl
label:
en_US: Stable Diffusion XL
- value: sd_3.5_large
label:
en_US: Stable Diffusion 3.5 Large
default: sd_3
label:
en_US: Choose Image Model