routers/api/v1/repo/modelarts.go

@@ -182,7 +182,7 @@ func GetModelArtsTrainJobVersion(ctx *context.APIContext) {
 		}
 		if oldStatus != job.Status {
 			notification.NotifyChangeCloudbrainStatus(job, oldStatus)
-			if models.IsTrainJobTerminal(job.Status) {
+			if models.IsTrainJobTerminal(job.Status) && job.ComputeResource == models.NPUResource {
 				if len(result.JobInfo.Tasks[0].CenterID) == 1 {
 					urchin.GetBackNpuModel(job.ID, grampus.GetRemoteEndPoint(result.JobInfo.Tasks[0].CenterID[0]), grampus.BucketRemote, grampus.GetNpuModelObjectKey(job.JobName), grampus.GetCenterProxy(setting.Grampus.LocalCenterID))
 				}

routers/repo/cloudbrain.go

@@ -1944,7 +1944,7 @@ func SyncCloudbrainStatus() {
 				task.CorrectCreateUnix()
 				if oldStatus != task.Status {
 					notification.NotifyChangeCloudbrainStatus(task, oldStatus)
-					if models.IsTrainJobTerminal(task.Status) {
+					if models.IsTrainJobTerminal(task.Status) && task.ComputeResource == models.NPUResource {
 						if len(result.JobInfo.Tasks[0].CenterID) == 1 {
 							urchin.GetBackNpuModel(task.ID, grampus.GetRemoteEndPoint(result.JobInfo.Tasks[0].CenterID[0]), grampus.BucketRemote, grampus.GetNpuModelObjectKey(task.JobName), grampus.GetCenterProxy(setting.Grampus.LocalCenterID))
 						}

routers/repo/grampus.go

@@ -879,7 +879,7 @@ func GrampusTrainJobShow(ctx *context.Context) {
 				task.CorrectCreateUnix()
 				if oldStatus != task.Status {
 					notification.NotifyChangeCloudbrainStatus(task, oldStatus)
-					if models.IsTrainJobTerminal(task.Status) {
+					if models.IsTrainJobTerminal(task.Status) && task.ComputeResource == models.NPUResource {
 						if len(result.JobInfo.Tasks[0].CenterID) == 1 {
 							urchin.GetBackNpuModel(task.ID, grampus.GetRemoteEndPoint(result.JobInfo.Tasks[0].CenterID[0]), grampus.BucketRemote, grampus.GetNpuModelObjectKey(task.JobName), grampus.GetCenterProxy(setting.Grampus.LocalCenterID))
 						}