From ecad3272d75d0b54eff992f1da2f00e61956362a Mon Sep 17 00:00:00 2001 From: zouap Date: Thu, 13 Oct 2022 17:49:20 +0800 Subject: [PATCH] =?UTF-8?q?=E5=87=BA=E9=94=99=E4=B9=8B=E5=90=8E=EF=BC=8C?= =?UTF-8?q?=E4=BB=BB=E5=8A=A1=E4=BF=A1=E6=81=AF=E5=8E=9F=E6=A0=B7=E8=BF=94?= =?UTF-8?q?=E5=9B=9E=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: zouap --- routers/repo/aisafety.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go index d4c162544..dde89dfab 100644 --- a/routers/repo/aisafety.go +++ b/routers/repo/aisafety.go @@ -978,6 +978,9 @@ func modelSafetyNewDataPrepare(ctx *context.Context) error { ctx.Data["CombatDataSetName"] = setting.ModelSafetyTest.GPUCombatDataSetName ctx.Data["CombatDataSetUUID"] = setting.ModelSafetyTest.GPUCombatDataSetUUID } else { + ctx.Data["engine_id"] = ctx.QueryInt("engine_id") + ctx.Data["pool_id"] = ctx.Query("pool_id") + ctx.Data["type"] = models.TypeCloudBrainTwo ctx.Data["compute_resource"] = models.NPUResource ctx.Data["datasetType"] = models.TypeCloudBrainTwo