diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go index f845c3918..eb3019e5a 100644 --- a/routers/repo/aisafety.go +++ b/routers/repo/aisafety.go @@ -351,6 +351,7 @@ func DelAiSafetyTask(ctx *context.Context) { func AiSafetyCreateForGetGPU(ctx *context.Context) { 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 @@ -362,6 +363,7 @@ func AiSafetyCreateForGetGPU(ctx *context.Context) { func AiSafetyCreateForGetNPU(ctx *context.Context) { ctx.Data["PageIsCloudBrain"] = true ctx.Data["IsCreate"] = true + ctx.Data["datasetType"] = models.TypeCloudBrainTwo 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">模型安全评测
@@ -168,7 +168,7 @@ {{.i18n.Tr "repo.cloudbrain.benchmark.model"}} 模型安全评测 + href="{{.RepoLink}}/modelsafety/create_gpu">模型安全评测
@@ -298,7 +298,7 @@ {{.i18n.Tr "repo.cloudbrain.benchmark.model"}} 模型安全评测 + href="{{.RepoLink}}/modelsafety/create_gpu">模型安全评测 diff --git a/templates/repo/modelsafety/indicators_descr.tmpl b/templates/repo/modelsafety/indicators_descr.tmpl new file mode 100644 index 000000000..3b6f7ff48 --- /dev/null +++ b/templates/repo/modelsafety/indicators_descr.tmpl @@ -0,0 +1,62 @@ + + + + + diff --git a/templates/repo/modelsafety/new.tmpl b/templates/repo/modelsafety/new.tmpl index b2d180215..1b6cccf46 100644 --- a/templates/repo/modelsafety/new.tmpl +++ b/templates/repo/modelsafety/new.tmpl @@ -49,19 +49,42 @@
-
+ {{.CsrfTokenHtml}} +
+
+
+ +
@@ -127,11 +150,18 @@ {{.i18n.Tr "cloudbrain.view_sample"}}
-
- +
+ + +
+
+ + +
- + - - - - + {{template "repo/modelsafety/indicators_descr".}}
@@ -199,7 +183,7 @@
-