From f7f4935564798ad64986fb47a89ea2bcbc814aaa Mon Sep 17 00:00:00 2001 From: chenyifan01 Date: Thu, 15 Dec 2022 11:10:53 +0800 Subject: [PATCH] #3277 gcu:update model selector --- modules/grampus/grampus.go | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) 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, }) }