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

Update relay-image.go

Browse Source
This commit is contained in:
Ghostz 2023-12-14 22:13:51 +08:00 committed by GitHub
parent 366b82128f
commit 64c8085b5e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
controller/relay-image.go
View File

@ -19,6 +19,9 @@ func isWithinRange(element string, value int) bool {
if _, ok := common.DalleGenerationImageAmounts[element]; !ok { if _, ok := common.DalleGenerationImageAmounts[element]; !ok {
return false return false
} }
if element == "dall-e-3" && value == 0 {
value = 1
}
min := common.DalleGenerationImageAmounts[element][0] min := common.DalleGenerationImageAmounts[element][0]
max := common.DalleGenerationImageAmounts[element][1] max := common.DalleGenerationImageAmounts[element][1]