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

Fixed suspect or

Browse Source
This commit is contained in:
ckt1031 2023-11-11 15:57:47 +08:00
parent 8361fbf4ef
commit e00bc7bd3f
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
controller/relay-image.go
View File

@ -6,11 +6,12 @@ import (
"encoding/json"
"errors"
"fmt"
"github.com/gin-gonic/gin"
"io"
"net/http"
"one-api/common"
"one-api/model"
"github.com/gin-gonic/gin"
)
func relayImageHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
@ -34,7 +35,7 @@ func relayImageHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode
}
// Model validation
if imageRequest.Model != "" || imageRequest.Model != "dall-e-3" {
if imageRequest.Model != "" && imageRequest.Model != "dall-e-3" {
imageModel = "dall-e-2"
}