From 7fead5b7d450f63d45b374ba9744d08e513ffb88 Mon Sep 17 00:00:00 2001 From: JustSong Date: Sun, 17 Dec 2023 18:52:28 +0800 Subject: [PATCH] fix: reset image num to 1 when not given --- controller/relay-image.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/controller/relay-image.go b/controller/relay-image.go index 288d4df7..bf916474 100644 --- a/controller/relay-image.go +++ b/controller/relay-image.go @@ -19,10 +19,6 @@ func isWithinRange(element string, value int) bool { if _, ok := common.DalleGenerationImageAmounts[element]; !ok { return false } - if element == "dall-e-3" && value == 0 { - value = 1 - } - min := common.DalleGenerationImageAmounts[element][0] max := common.DalleGenerationImageAmounts[element][1] @@ -45,6 +41,10 @@ func relayImageHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode return errorWrapper(err, "bind_request_body_failed", http.StatusBadRequest) } + if imageRequest.N == 0 { + imageRequest.N = 1 + } + // Size validation if imageRequest.Size != "" { imageSize = imageRequest.Size