Browse Source

提交代码

tags/v1.22.12.1^2
ychao_1983 3 years ago
parent
commit
775db3b9a8
2 changed files with 0 additions and 18 deletions
  1. +0
    -10
      modules/grampus/grampus.go
  2. +0
    -8
      routers/repo/grampus.go

+ 0
- 10
modules/grampus/grampus.go View File

@@ -188,16 +188,6 @@ func GenerateNotebookJob(ctx *context.Context, req *GenerateNotebookJobReq) (job
ContainerPath: cloudbrain.PretrainModelMountPath,
})
}

datasetGrampus = append(datasetGrampus, models.GrampusDataset{
Name: "modelOutput",
Bucket: setting.Attachment.Minio.Bucket,
EndPoint: setting.Attachment.Minio.Endpoint,
ReadOnly: false,
ObjectKey: req.ModelPath,
ContainerPath: cloudbrain.ModelMountPath,
})

codeGrampus = models.GrampusDataset{
Name: req.CodeName,
Bucket: setting.Attachment.Minio.Bucket,


+ 0
- 8
routers/repo/grampus.go View File

@@ -106,7 +106,6 @@ func GrampusNotebookCreate(ctx *context.Context, form auth.CreateGrampusNotebook
image := strings.TrimSpace(form.Image)

codeStoragePath := setting.CBCodePathPrefix + jobName + cloudbrain.CodeMountPath + "/"
modelStoragePath := setting.CBCodePathPrefix + jobName + cloudbrain.ModelMountPath + "/"

tpl := tplGrampusNotebookGPUNew
processType := grampus.ProcessorTypeGPU
@@ -223,12 +222,6 @@ func GrampusNotebookCreate(ctx *context.Context, form auth.CreateGrampusNotebook
ctx.RenderWithErr(ctx.Tr("cloudbrain.load_code_failed"), tpl, &form)
return
}
errStr := initModelPath(jobName, cloudbrain.ModelMountPath)
if errStr != "" {
grampusNotebookNewDataPrepare(ctx, processType)
ctx.RenderWithErr(ctx.Tr(errStr), tpl, &form)
return
}

} else {

@@ -258,7 +251,6 @@ func GrampusNotebookCreate(ctx *context.Context, form auth.CreateGrampusNotebook
Spec: spec,
CodeStoragePath: codeStoragePath,
CodeName: strings.ToLower(repo.Name),
ModelPath: modelStoragePath + README,
}

if form.ModelName != "" { //使用预训练模型训练


Loading…
Cancel
Save