diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index 25a76fa41..43f4a6e73 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -294,6 +294,9 @@ func NotebookShow(ctx *context.Context) {
 		} else { //deleted record
 
 		}
+		if task.FlavorCode == "" {
+			task.FlavorCode = result.Flavor
+		}
 	}
 
 	datasetDownload := make([]models.DatasetDownload, 0)
@@ -528,6 +531,8 @@ func NotebookManage(ctx *context.Context) {
 				Description:     task.Description,
 				CreatedUnix:     createTime,
 				UpdatedUnix:     createTime,
+				FlavorCode:      task.FlavorCode,
+				FlavorName:      task.FlavorName,
 			}
 
 			err = models.RestartCloudbrain(task, newTask)