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 {
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