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

11
controller/relay-image.go
View File

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