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 {