Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Support Ali stable-diffusion-xl and wanx-v1 model

Browse Source
This commit is contained in:
mo 2024-03-29 16:03:29 +08:00
parent e99b5ebd46
commit b125358464
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
relay/constant/image.go
View File

@ -26,15 +26,13 @@ var DalleSizeRatios = map[string]map[string]float64{
} }
var DalleGenerationImageAmounts = map[string][2]int{ var DalleGenerationImageAmounts = map[string][2]int{
"dall-e-2": {1, 10}, "dall-e-2": {1, 10},
"dall-e-3": {1, 1}, // OpenAI allows n=1 currently. "dall-e-3": {1, 1}, // OpenAI allows n=1 currently.
"stable-diffusion-xl": {1, 4}, // Ali "stable-diffusion-xl": {1, 4}, // Ali
"wanx-v1": {1, 4}, // Ali "wanx-v1": {1, 4}, // Ali
} }
var DalleImagePromptLengthLimitations = map[string]int{ var DalleImagePromptLengthLimitations = map[string]int{
"dall-e-2": 1000,
"dall-e-3": 4000,
"dall-e-2": 1000, "dall-e-2": 1000,
"dall-e-3": 4000, "dall-e-3": 4000,
"stable-diffusion-xl": 4000, "stable-diffusion-xl": 4000,