Browse Source

只能挂载文件

tags/v1.22.12.1^2
ychao_1983 3 years ago
parent
commit
ad3627c44c
2 changed files with 3 additions and 2 deletions
  1. +2
    -1
      routers/repo/cloudbrain.go
  2. +1
    -1
      routers/repo/grampus.go

+ 2
- 1
routers/repo/cloudbrain.go View File

@@ -81,6 +81,7 @@ var (


const BENCHMARK_TYPE_CODE = "repo.cloudbrain.benchmark.types" const BENCHMARK_TYPE_CODE = "repo.cloudbrain.benchmark.types"
const CLONE_FILE_PREFIX = "file:///" const CLONE_FILE_PREFIX = "file:///"
const README = "README"


var benchmarkTypesMap = make(map[string]*models.BenchmarkTypes, 0) var benchmarkTypesMap = make(map[string]*models.BenchmarkTypes, 0)


@@ -1767,7 +1768,7 @@ func mkPathAndReadMeFile(path string, text string) error {
return err return err
} }


fileName := path + "README"
fileName := path + README
f, err := os.OpenFile(fileName, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, os.ModePerm) f, err := os.OpenFile(fileName, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, os.ModePerm)
if err != nil { if err != nil {
log.Error("OpenFile failed", err.Error()) log.Error("OpenFile failed", err.Error())


+ 1
- 1
routers/repo/grampus.go View File

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


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


Loading…
Cancel
Save