diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go index 3b642d2991..04eb7531de 100755 --- a/routers/repo/cloudbrain.go +++ b/routers/repo/cloudbrain.go @@ -1362,7 +1362,7 @@ func StopJobs(cloudBrains []*models.Cloudbrain) { logErrorAndUpdateJobStatus(err, taskInfo) } else if taskInfo.Type == models.TypeCloudBrainTwo { - if taskInfo.JobType == string(models.JobTypeTrain) { + if taskInfo.JobType == string(models.JobTypeTrain) || taskInfo.JobType == string(models.JobTypeInference) { err := retry(3, time.Second*30, func() error { _, err := modelarts.StopTrainJob(taskInfo.JobID, strconv.FormatInt(taskInfo.VersionID, 10)) return err