From c0888e04830c483a8ad07be2241e0cadee4d5e3d Mon Sep 17 00:00:00 2001 From: chenyifan01 Date: Wed, 14 Dec 2022 11:38:34 +0800 Subject: [PATCH] #3364 repo-square:fix bug --- routers/repo/ai_model_manage.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go index b55c92885..a45f21cca 100644 --- a/routers/repo/ai_model_manage.go +++ b/routers/repo/ai_model_manage.go @@ -171,12 +171,12 @@ func updateStatus(id string, modelSize int64, status int, modelPath string, stat if len(statusDesc) > 400 { statusDesc = statusDesc[0:400] } + m, _ := models.QueryModelById(id) err := models.ModifyModelStatus(id, modelSize, status, modelPath, statusDesc) if err != nil { log.Info("update status error." + err.Error()) } - m, err := models.QueryModelById(id) - if err == nil { + if m != nil { if modelSize > 0 && m.Size == 0 { go repository.IncreaseRepoModelNum(m.RepoId) }