#1249

fix bug

models/action.go

@@ -384,7 +384,9 @@ func (a *Action) IsCloudbrainAction() bool {
 		ActionCreateTrainTask,
 		ActionCreateInferenceTask,
 		ActionCreateBenchMarkTask,
-		ActionCreateGPUTrainTask:
+		ActionCreateGPUTrainTask,
+		ActionCreateGrampusNPUTrainTask,
+		ActionCreateGrampusGPUTrainTask:
 		return true
 	}
 	return false