#3364

repo-square:fix bug

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) {