From 83e6243bacb389106d38223141af7694ed046755 Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Thu, 22 Dec 2022 16:39:45 +0800
Subject: [PATCH] fix-bug

---
 services/cloudbrain/cloudbrainTask/notebook.go | 3 +++
 1 file changed, 3 insertions(+)

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
 }