From 8f37e3baf441ea031b292529be36f8439dc17f7d Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Mon, 9 Jan 2023 09:47:03 +0800
Subject: [PATCH] fix-3459

---
 services/cloudbrain/cloudbrainTask/notebook.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/services/cloudbrain/cloudbrainTask/notebook.go b/services/cloudbrain/cloudbrainTask/notebook.go
index 3526f6549..83024b81c 100644
--- a/services/cloudbrain/cloudbrainTask/notebook.go
+++ b/services/cloudbrain/cloudbrainTask/notebook.go
@@ -417,7 +417,7 @@ func cloudBrainFileNoteBookCreate(ctx *context.Context, option api.CreateFileNot
 }
 
 func getCloudbrainType(optionType int) int {
-	if optionType < 1 {
+	if optionType <= GPUType {
 		return models.TypeCloudBrainOne
 	}
 	if setting.ModelartsCD.Enabled {