From 1768d1ef558bb6068af4079aa720cee1f26ca93c Mon Sep 17 00:00:00 2001
From: ckt1031 <65409152+ckt1031@users.noreply.github.com>
Date: Sat, 11 Nov 2023 16:03:30 +0800
Subject: [PATCH] Refactored size ratio calculation in relayImageHelper
 function

---
 controller/relay-image.go | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/controller/relay-image.go b/controller/relay-image.go
index 30fdadf5..2f2e57cf 100644
--- a/controller/relay-image.go
+++ b/controller/relay-image.go
@@ -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 {
-			sizeRatio = ratio
-
-			if imageRequest.Quality == "hd" {
-				sizeRatio = ratio * 2
+	if ratio, ok := common.DalleSizeRatios[imageModel][requestSize]; ok {
+		sizeRatio = ratio
+		if imageRequest.Quality == "hd" && imageRequest.Model == "dall-e-3" {
+			if requestSize == "1024x1024" {
+				sizeRatio *= 2
+			} else {
+				sizeRatio *= 1.5
 			}
 		}
 	}