diff --git a/services/cloudbrain/cloudbrainTask/notebook.go b/services/cloudbrain/cloudbrainTask/notebook.go
index 4a06bbc79..3526f6549 100644
--- a/services/cloudbrain/cloudbrainTask/notebook.go
+++ b/services/cloudbrain/cloudbrainTask/notebook.go
@@ -420,6 +420,9 @@ func getCloudbrainType(optionType int) int {
 	if optionType < 1 {
 		return models.TypeCloudBrainOne
 	}
+	if setting.ModelartsCD.Enabled {
+		return models.TypeCDCenter
+	}
 	return models.TypeCloudBrainTwo
 }