diff --git a/models/ai_model_manage.go b/models/ai_model_manage.go index 9c29b6c7a..a210124a4 100644 --- a/models/ai_model_manage.go +++ b/models/ai_model_manage.go @@ -120,15 +120,11 @@ func ModifyModelNewProperty(id string, new int, versioncount int) error { func ModifyModelDownloadCount(id string) error { var sess *xorm.Session defer sess.Close() - if err := sess.Begin(); err != nil { - return err - } + if _, err := sess.Exec("UPDATE `ai_model_manage` SET download_count = download_count + 1 WHERE id = ?", id); err != nil { return err } - if err := sess.Commit(); err != nil { - return err - } + return nil }