|
|
@@ -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, |
|
|
|
}) |
|
|
|
} |
|
|
|
|
|
|
|