From cc9d7adc72ac176ae0f91756523e0c089fa491c0 Mon Sep 17 00:00:00 2001
From: chenyifan01 <cyf_uestc@outlook.com>
Date: Wed, 14 Sep 2022 15:38:54 +0800
Subject: [PATCH] #1249 fix bug

---
 models/action.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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