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

Refactored size ratio calculation in

Browse Source 
relayImageHelper function
This commit is contained in:
ckt1031 2023-11-11 16:03:30 +08:00
parent e00bc7bd3f
commit 1768d1ef55
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
controller/relay-image.go
View File

@ -87,12 +87,13 @@ func relayImageHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode
sizeRatio := 1.0 sizeRatio := 1.0
if ratios, ok := common.DalleSizeRatios[imageModel]; ok { if ratio, ok := common.DalleSizeRatios[imageModel][requestSize]; ok {
if ratio, ok := ratios[requestSize]; ok { sizeRatio = ratio
sizeRatio = ratio if imageRequest.Quality == "hd" && imageRequest.Model == "dall-e-3" {
if requestSize == "1024x1024" {
if imageRequest.Quality == "hd" { sizeRatio *= 2
sizeRatio = ratio * 2 } else {
sizeRatio *= 1.5
} }
} }
} }