Browse Source

#3277

gcu:update model selector
gcu
chenyifan01 2 years ago
parent
commit
f7f4935564
1 changed files with 6 additions and 5 deletions
  1. +6
    -5
      modules/grampus/grampus.go

+ 6
- 5
modules/grampus/grampus.go View File

@@ -227,11 +227,12 @@ func GenerateNotebookJob(ctx *context.Context, req *GenerateNotebookJobReq) (job
}) })
} else { } else {
datasetGrampus = append(datasetGrampus, models.GrampusDataset{ 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,
}) })
} }




Loading…
Cancel
Save