xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-04 16:23:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: fix bug for issue 49, stable diffusion service not decrease img_calls

Browse Source
This commit is contained in:
RockYang
2023-11-11 11:00:26 +08:00
parent 7b4ef8fc31
commit 2525a22d78
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/service/sd/service.go
View File

@@ -287,7 +287,8 @@ func (s *Service) callback(data CBReq) {
if data.Progress < 100 && data.ImageData != "" { if data.Progress < 100 && data.ImageData != "" {
jobVo.ImgURL = data.ImageData jobVo.ImgURL = data.ImageData
} }
// 扣减绘图次数
s.db.Where("id = ?", jobVo.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls - ?", 1))
// 推送任务到前端 // 推送任务到前端
if client != nil { if client != nil {
utils.ReplyChunkMessage(client, jobVo) utils.ReplyChunkMessage(client, jobVo)