From 72150f8f03abde8c63253a6a3ffdfffaae8def3f Mon Sep 17 00:00:00 2001
From: chenyifan01 <cyf_uestc@outlook.com>
Date: Wed, 14 Dec 2022 10:24:04 +0800
Subject: [PATCH] #3364 repo-square:fix bug

---
 routers/repo/ai_model_manage.go | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go
index 605284430..b55c92885 100644
--- a/routers/repo/ai_model_manage.go
+++ b/routers/repo/ai_model_manage.go
@@ -122,9 +122,6 @@ func saveModelByParameters(jobId string, versionName string, name string, versio
 	if err != nil {
 		return "", err
 	}
-	if modelSize > 0 {
-		go repository.IncreaseRepoModelNum(aiTask.RepoID)
-	}
 	if len(lastNewModelId) > 0 {
 		//udpate status and version count
 		models.ModifyModelNewProperty(lastNewModelId, MODEL_NOT_LATEST, 0)
@@ -178,6 +175,13 @@ func updateStatus(id string, modelSize int64, status int, modelPath string, stat
 	if err != nil {
 		log.Info("update status error." + err.Error())
 	}
+	m, err := models.QueryModelById(id)
+	if err == nil {
+		if modelSize > 0 && m.Size == 0 {
+			go repository.IncreaseRepoModelNum(m.RepoId)
+		}
+	}
+
 }
 
 func SaveNewNameModel(ctx *context.Context) {