diff --git a/models/action.go b/models/action.go
index 79f980a93..db3114c95 100755
--- a/models/action.go
+++ b/models/action.go
@@ -384,7 +384,9 @@ func (a *Action) IsCloudbrainAction() bool {
 		ActionCreateTrainTask,
 		ActionCreateInferenceTask,
 		ActionCreateBenchMarkTask,
-		ActionCreateGPUTrainTask:
+		ActionCreateGPUTrainTask,
+		ActionCreateGrampusNPUTrainTask,
+		ActionCreateGrampusGPUTrainTask:
 		return true
 	}
 	return false