fix: "Inpaint" code error

da55c56c · CaIon · 6ec45751 · da55c56c
Commit da55c56c authored Mar 13, 2024 by CaIon
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

relay/relay-mj.go
+4 -2

No files found.
--- a/relay/relay-mj.go
+++ b/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()