test/geekai
1
0
Fork 0
mirror of https://gitee.com/blackfox/geekai.git synced 2024-12-02 20:28:00 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: Upscale and Variation task overrite each other

Browse Source
This commit is contained in:
RockYang 2024-02-16 18:08:29 +08:00
parent 8ab15e5dc4
commit b0162e6a92
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/service/mj/service.go
View File

@ -137,6 +137,12 @@ func (s *Service) Notify(data CBReq) {
} else {
tx = tx.Where("task_id = ?", split[0])
}
// fixed: 修复 U/V 操作任务混淆覆盖的 Bug
if strings.Contains(data.Prompt, "** - Image #") { // for upscale
tx = tx.Where("type = ?", types.TaskUpscale.String())
} else if strings.Contains(data.Prompt, "** - Variations (Strong)") { // for Variations
tx = tx.Where("type = ?", types.TaskVariation.String())
}
res = tx.First(&job)
if res.Error != nil {
logger.Warn("非法任务：", res.Error)