Explorar el Código

fix: "Inpaint" code error

CaIon hace 2 años
padre
commit
44361d75e8
Se han modificado 1 ficheros con 4 adiciones y 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()