Explorar o código

fix: "Inpaint" code error

CaIon %!s(int64=2) %!d(string=hai) anos
pai
achega
44361d75e8
Modificáronse 1 ficheiros con 4 adicións e 2 borrados
  1. 4 2
      relay/relay-mj.go

+ 4 - 2
relay/relay-mj.go

@@ -494,8 +494,10 @@ func RelayMidjourneySubmit(c *gin.Context, relayMode int) *dto.MidjourneyRespons
 			}
 		}
 		//修改返回值
-		newBody := strings.Replace(string(responseBody), `"code":21`, `"code":1`, -1)
-		responseBody = []byte(newBody)
+		if midjRequest.Action != constant.MjActionInPaintPre {
+			newBody := strings.Replace(string(responseBody), `"code":21`, `"code":1`, -1)
+			responseBody = []byte(newBody)
+		}
 	}
 
 	err = midjourneyTask.Insert()