From 3df7cddfbf36e5d6d897d1510349190ccb95d8e4 Mon Sep 17 00:00:00 2001 From: liuzx Date: Tue, 31 May 2022 14:46:07 +0800 Subject: [PATCH] fix-1941 --- models/cloudbrain.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/models/cloudbrain.go b/models/cloudbrain.go index c1f346c89..810e68d30 100755 --- a/models/cloudbrain.go +++ b/models/cloudbrain.go @@ -1567,7 +1567,10 @@ func GetCloudbrainCountByUserID(userID int64, jobType string) (int, error) { } func GetCloudbrainRunCountByRepoID(repoID int64) (int, error) { - count, err := x.In("status", JobWaiting, JobRunning).And("repo_id = ?", repoID).Count(new(Cloudbrain)) + count, err := x.In("status", JobWaiting, JobRunning, ModelArtsCreateQueue, ModelArtsCreating, ModelArtsStarting, + ModelArtsReadyToStart, ModelArtsResizing, ModelArtsStartQueuing, ModelArtsRunning, ModelArtsRestarting, ModelArtsTrainJobInit, + ModelArtsTrainJobImageCreating, ModelArtsTrainJobSubmitTrying, ModelArtsTrainJobWaiting, ModelArtsTrainJobRunning, + ModelArtsTrainJobScaling, ModelArtsTrainJobCheckInit, ModelArtsTrainJobCheckRunning, ModelArtsTrainJobCheckRunningCompleted).And("repo_id = ?", repoID).Count(new(Cloudbrain)) return int(count), err }