diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go index 9b7af5d6c..30cb50c79 100644 --- a/routers/repo/aisafety.go +++ b/routers/repo/aisafety.go @@ -526,6 +526,8 @@ func AiSafetyCreateForGetGPU(ctx *context.Context) { func AiSafetyCreateForGetGrampusGPU(ctx *context.Context) { ctx.Data["PageIsCloudBrain"] = true ctx.Data["IsCreate"] = true + ctx.Data["type"] = models.TypeC2Net + ctx.Data["compute_resource"] = models.GPUResource ctx.Data["datasetType"] = models.TypeCloudBrainOne ctx.Data["BaseDataSetName"] = setting.ModelSafetyTest.BaseDataSetName ctx.Data["BaseDataSetUUID"] = setting.ModelSafetyTest.BaseDataSetUUID @@ -542,7 +544,8 @@ func AiSafetyCreateForGetGrampusGPU(ctx *context.Context) { func AiSafetyCreateForGetGrampusNPU(ctx *context.Context) { ctx.Data["PageIsCloudBrain"] = true ctx.Data["IsCreate"] = true - + ctx.Data["type"] = models.TypeC2Net + ctx.Data["compute_resource"] = models.NPUResource ctx.Data["datasetType"] = models.TypeCloudBrainTwo ctx.Data["BaseDataSetName"] = setting.ModelSafetyTest.BaseDataSetName ctx.Data["BaseDataSetUUID"] = setting.ModelSafetyTest.BaseDataSetUUID @@ -678,7 +681,7 @@ func createForGrampusGPU(ctx *context.Context, jobName string) { image := strings.TrimSpace(ctx.Query("image")) srcDataset := ctx.Query("src_dataset") //uuid combatDataset := ctx.Query("combat_dataset") //uuid - evaluationIndex := ctx.Query("evaluationIndex") + evaluationIndex := ctx.Query("evaluation_index") Params := ctx.Query("run_para_list") specId := ctx.QueryInt64("spec_id") TrainUrl := ctx.Query("train_url") @@ -828,7 +831,7 @@ func createForNPU(ctx *context.Context, jobName string) { srcDataset := ctx.Query("src_dataset") //uuid combatDataset := ctx.Query("combat_dataset") //uuid - evaluationIndex := ctx.Query("evaluationIndex") + evaluationIndex := ctx.Query("evaluation_index") Params := ctx.Query("run_para_list") specId := ctx.QueryInt64("spec_id") @@ -1018,7 +1021,7 @@ func createForGPU(ctx *context.Context, jobName string) { image := strings.TrimSpace(ctx.Query("image")) srcDataset := ctx.Query("src_dataset") //uuid combatDataset := ctx.Query("combat_dataset") //uuid - evaluationIndex := ctx.Query("evaluationIndex") + evaluationIndex := ctx.Query("evaluation_index") Params := ctx.Query("run_para_list") specId := ctx.QueryInt64("spec_id") TrainUrl := ctx.Query("train_url") @@ -1146,7 +1149,7 @@ func modelSafetyNewDataPrepare(ctx *context.Context) error { ctx.Data["image"] = strings.TrimSpace(ctx.Query("image")) ctx.Data["src_dataset"] = ctx.Query("src_dataset") //uuid ctx.Data["combat_dataset"] = ctx.Query("combat_dataset") //uuid - ctx.Data["evaluationIndex"] = ctx.Query("evaluationIndex") + ctx.Data["evaluationIndex"] = ctx.Query("evaluation_index") ctx.Data["run_para_list"] = ctx.Query("run_para_list") ctx.Data["spec_id"] = ctx.QueryInt64("spec_id") ctx.Data["train_url"] = ctx.Query("train_url")