From cf7056e015d038b7d6fcdb30bd9b42f300a633ca Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Wed, 14 Dec 2022 10:41:16 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 routers/repo/grampus.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/routers/repo/grampus.go b/routers/repo/grampus.go
index f471f0c4d..df56cfa94 100755
--- a/routers/repo/grampus.go
+++ b/routers/repo/grampus.go
@@ -1696,6 +1696,11 @@ func GrampusNotebookRestart(ctx *context.Context) {
 			Type:             task.Type,
 			Uuid:             task.Uuid,
 			Image:            task.Image,
+			ImageID:          task.ImageID,
+			EngineID:         task.EngineID,
+			CommitID:         task.CommitID,
+			EngineName:       task.EngineName,
+			IsLatestVersion:  "1",
 			BranchName:       task.BranchName,
 			DatasetName:      task.DatasetName,
 			ComputeResource:  task.ComputeResource,
@@ -1708,6 +1713,7 @@ func GrampusNotebookRestart(ctx *context.Context) {
 			LabelName:        task.LabelName,
 			PreTrainModelUrl: task.PreTrainModelUrl,
 			CkptName:         task.CkptName,
+			WorkServerNumber: 1,
 		}
 
 		err = models.RestartCloudbrain(task, newTask)