diff --git a/models/cloudbrain.go b/models/cloudbrain.go index c1c01b581..dc11d3459 100755 --- a/models/cloudbrain.go +++ b/models/cloudbrain.go @@ -1889,7 +1889,7 @@ func GetCloudbrainCountByUserID(userID int64, jobType string) (int, error) { func GetCloudbrainRunCountByRepoID(repoID int64) (int, error) { count, err := x.In("status", JobWaiting, JobRunning, ModelArtsCreateQueue, ModelArtsCreating, ModelArtsStarting, ModelArtsReadyToStart, ModelArtsResizing, ModelArtsStartQueuing, ModelArtsRunning, ModelArtsRestarting, ModelArtsTrainJobInit, - ModelArtsTrainJobImageCreating, ModelArtsTrainJobSubmitTrying, ModelArtsTrainJobWaiting, ModelArtsTrainJobRunning, + ModelArtsTrainJobImageCreating, ModelArtsTrainJobSubmitTrying, ModelArtsTrainJobWaiting, ModelArtsTrainJobRunning, ModelArtsStopping, ModelArtsResizing, ModelArtsTrainJobScaling, ModelArtsTrainJobCheckInit, ModelArtsTrainJobCheckRunning, ModelArtsTrainJobCheckRunningCompleted).And("repo_id = ?", repoID).Count(new(Cloudbrain)) return int(count), err } diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go index cd5f7430a..457f275ed 100755 --- a/routers/repo/cloudbrain.go +++ b/routers/repo/cloudbrain.go @@ -1251,7 +1251,6 @@ func DeleteJobsByRepoID(repoID int64) { log.Warn("Failed to get cloudBrain info", err) return } - time.Sleep(time.Second * 120) DeleteJobs(cloudBrains) }