diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go index 179cb8cc4..4d227fc9d 100644 --- a/routers/repo/aisafety.go +++ b/routers/repo/aisafety.go @@ -352,6 +352,7 @@ func AiSafetyCreateForGetGPU(ctx *context.Context) { t := time.Now() ctx.Data["PageIsCloudBrain"] = true ctx.Data["IsCreate"] = true + ctx.Data["datasetType"] = models.TypeCloudBrainOne ctx.Data["BaseDataSetName"] = setting.ModelSafetyTest.BaseDataSetName ctx.Data["BaseDataSetUUID"] = setting.ModelSafetyTest.BaseDataSetUUID ctx.Data["CombatDataSetName"] = setting.ModelSafetyTest.CombatDataSetName @@ -370,8 +371,12 @@ func AiSafetyCreateForGetNPU(ctx *context.Context) { t := time.Now() ctx.Data["PageIsCloudBrain"] = true ctx.Data["IsCreate"] = true +<<<<<<< HEAD var displayJobName = jobNamePrefixValid(cutString(ctx.User.Name, 5)) + t.Format("2006010215") + strconv.Itoa(int(t.Unix()))[5:] ctx.Data["display_job_name"] = displayJobName +======= + ctx.Data["datasetType"] = models.TypeCloudBrainTwo +>>>>>>> 6cd7f96040b300f1a10b1423312fee373511bc07 ctx.Data["BaseDataSetName"] = setting.ModelSafetyTest.BaseDataSetName ctx.Data["BaseDataSetUUID"] = setting.ModelSafetyTest.BaseDataSetUUID ctx.Data["CombatDataSetName"] = setting.ModelSafetyTest.CombatDataSetName diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go index 1f6732174..b1020c082 100755 --- a/routers/repo/cloudbrain.go +++ b/routers/repo/cloudbrain.go @@ -51,8 +51,8 @@ const ( tplCloudBrainBenchmarkNew base.TplName = "repo/cloudbrain/benchmark/new" tplCloudBrainBenchmarkShow base.TplName = "repo/cloudbrain/benchmark/show" - tplCloudBrainModelSafetyNewGpu base.TplName = "repo/cloudbrain/modelsafety/newgpu" - tplCloudBrainModelSafetyNewNpu base.TplName = "repo/cloudbrain/modelsafety/newnpu" + tplCloudBrainModelSafetyNewGpu base.TplName = "repo/modelsafety/newgpu" + tplCloudBrainModelSafetyNewNpu base.TplName = "repo/modelsafety/newnpu" tplCloudBrainImageSubmit base.TplName = "repo/cloudbrain/image/submit" tplCloudBrainImageEdit base.TplName = "repo/cloudbrain/image/edit" diff --git a/templates/repo/cloudbrain/benchmark/new.tmpl b/templates/repo/cloudbrain/benchmark/new.tmpl index 863edd2d0..13665c036 100755 --- a/templates/repo/cloudbrain/benchmark/new.tmpl +++ b/templates/repo/cloudbrain/benchmark/new.tmpl @@ -51,7 +51,7 @@ {{.i18n.Tr "repo.cloudbrain.benchmark.model"}} 模型安全评测 + href="{{.RepoLink}}/modelsafety/create_gpu">模型安全评测