diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index 0dc299a3b..fe65386cb 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -528,6 +528,8 @@ func NotebookManage(ctx *context.Context) {
 				Description:     task.Description,
 				CreatedUnix:     createTime,
 				UpdatedUnix:     createTime,
+				FlavorName:      task.FlavorName,
+				FlavorCode:      task.FlavorCode,
 			}
 
 			err = models.RestartCloudbrain(task, newTask)