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

Refactored variable name from

Browse Source 
DalleGenerationImageAmountRatios to
DalleGenerationImageAmounts
This commit is contained in:
ckt1031 2023-11-15 20:13:48 +08:00
parent 6f5655ec7a
commit 29fbd12870
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
common/model-ratio.go
View File

@ -19,7 +19,7 @@ var DalleSizeRatios = map[string]map[string]float64{
} }
// Array in values, of maxmium and minimum // Array in values, of maxmium and minimum
var DalleGenerationImageAmountRatios = 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.
} }

6
controller/relay-image.go
View File

@ -15,12 +15,12 @@ import (
) )
func isWithinRange(element string, value int) bool { func isWithinRange(element string, value int) bool {
if _, ok := common.DalleGenerationImageAmountRatios[element]; !ok { if _, ok := common.DalleGenerationImageAmounts[element]; !ok {
return false return false
} }
min := common.DalleGenerationImageAmountRatios[element][0] min := common.DalleGenerationImageAmounts[element][0]
max := common.DalleGenerationImageAmountRatios[element][1] max := common.DalleGenerationImageAmounts[element][1]
return value >= min && value <= max return value >= min && value <= max
} }