diff --git a/modules/grampus/grampus.go b/modules/grampus/grampus.go
index c55fd58ea..d82dd018f 100755
--- a/modules/grampus/grampus.go
+++ b/modules/grampus/grampus.go
@@ -227,11 +227,12 @@ func GenerateNotebookJob(ctx *context.Context, req *GenerateNotebookJobReq) (job
 				})
 			} else {
 				datasetGrampus = append(datasetGrampus, models.GrampusDataset{
-					Name:      req.ModelName,
-					Bucket:    setting.Bucket,
-					EndPoint:  getEndPoint(),
-					ReadOnly:  true,
-					ObjectKey: req.PreTrainModelPath,
+					Name:          req.ModelName,
+					Bucket:        setting.Bucket,
+					EndPoint:      getEndPoint(),
+					ReadOnly:      true,
+					ObjectKey:     req.PreTrainModelPath,
+					ContainerPath: cloudbrain.PretrainModelMountPath,
 				})
 			}