diff --git a/routers/repo/grampus.go b/routers/repo/grampus.go
index 066f29585..48957387f 100755
--- a/routers/repo/grampus.go
+++ b/routers/repo/grampus.go
@@ -89,6 +89,12 @@ func grampusTrainJobNewDataPrepare(ctx *context.Context, processType string) err
 
 	ctx.Data["branchName"] = ctx.Repo.BranchName
 
+	if processType == grampus.ProcessorTypeGPU {
+		ctx.Data["datasetType"] = models.TypeCloudBrainOne
+	} else if processType == grampus.ProcessorTypeNPU {
+		ctx.Data["datasetType"] = models.TypeCloudBrainTwo
+	}
+
 	return nil
 }