Bladeren bron

fix: "Inpaint" code error

CaIon 2 jaren geleden
bovenliggende
commit
44361d75e8
1 gewijzigde bestanden met toevoegingen van 4 en 2 verwijderingen
  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()