diff --git a/relay/relay-image.go b/relay/relay-image.go index 74d6c301..411a2c39 100644 --- a/relay/relay-image.go +++ b/relay/relay-image.go @@ -38,9 +38,7 @@ func getAndValidImageRequest(c *gin.Context, info *relaycommon.RelayInfo) (*dto. if imageRequest.Model == "" { imageRequest.Model = "dall-e-2" } - if imageRequest.Quality == "" { - imageRequest.Quality = "standard" - } + // Not "256x256", "512x512", or "1024x1024" if imageRequest.Model == "dall-e-2" || imageRequest.Model == "dall-e" { if imageRequest.Size != "" && imageRequest.Size != "256x256" && imageRequest.Size != "512x512" && imageRequest.Size != "1024x1024" { @@ -50,6 +48,9 @@ func getAndValidImageRequest(c *gin.Context, info *relaycommon.RelayInfo) (*dto. if imageRequest.Size != "" && imageRequest.Size != "1024x1024" && imageRequest.Size != "1024x1792" && imageRequest.Size != "1792x1024" { return nil, errors.New("size must be one of 256x256, 512x512, or 1024x1024, dall-e-3 1024x1792 or 1792x1024") } + if imageRequest.Quality == "" { + imageRequest.Quality = "standard" + } //if imageRequest.N != 1 { // return nil, errors.New("n must be 1") //}