1 changed files with 8 additions and 0 deletions
Unified View
Diff Options
-
+8 -0routers/repo/aisafety.go
+ 8
- 0
routers/repo/aisafety.go
View File
| @@ -526,6 +526,10 @@ func AiSafetyCreateForGetGPU(ctx *context.Context) { | |||||
| ctx.Data["BaseDataSetUUID"] = setting.ModelSafetyTest.GPUBaseDataSetUUID | ctx.Data["BaseDataSetUUID"] = setting.ModelSafetyTest.GPUBaseDataSetUUID | ||||
| ctx.Data["CombatDataSetName"] = setting.ModelSafetyTest.GPUCombatDataSetName | ctx.Data["CombatDataSetName"] = setting.ModelSafetyTest.GPUCombatDataSetName | ||||
| ctx.Data["CombatDataSetUUID"] = setting.ModelSafetyTest.GPUCombatDataSetUUID | ctx.Data["CombatDataSetUUID"] = setting.ModelSafetyTest.GPUCombatDataSetUUID | ||||
| log.Info("GPUBaseDataSetName=" + setting.ModelSafetyTest.GPUBaseDataSetName) | |||||
| log.Info("GPUBaseDataSetUUID=" + setting.ModelSafetyTest.GPUBaseDataSetUUID) | |||||
| log.Info("GPUCombatDataSetName=" + setting.ModelSafetyTest.GPUCombatDataSetName) | |||||
| log.Info("GPUCombatDataSetUUID=" + setting.ModelSafetyTest.GPUCombatDataSetUUID) | |||||
| var displayJobName = jobNamePrefixValid(cutString(ctx.User.Name, 5)) + t.Format("2006010215") + strconv.Itoa(int(t.Unix()))[5:] | 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["display_job_name"] = displayJobName | ||||
| prepareCloudbrainOneSpecs(ctx) | prepareCloudbrainOneSpecs(ctx) | ||||
| @@ -550,6 +554,10 @@ func AiSafetyCreateForGetNPU(ctx *context.Context) { | |||||
| ctx.Data["CombatDataSetName"] = setting.ModelSafetyTest.NPUCombatDataSetName | ctx.Data["CombatDataSetName"] = setting.ModelSafetyTest.NPUCombatDataSetName | ||||
| ctx.Data["CombatDataSetUUID"] = setting.ModelSafetyTest.NPUCombatDataSetUUID | ctx.Data["CombatDataSetUUID"] = setting.ModelSafetyTest.NPUCombatDataSetUUID | ||||
| log.Info("NPUBaseDataSetName=" + setting.ModelSafetyTest.NPUBaseDataSetName) | |||||
| log.Info("NPUBaseDataSetUUID=" + setting.ModelSafetyTest.NPUBaseDataSetUUID) | |||||
| log.Info("NPUCombatDataSetName=" + setting.ModelSafetyTest.NPUCombatDataSetName) | |||||
| log.Info("NPUCombatDataSetUUID=" + setting.ModelSafetyTest.NPUCombatDataSetUUID) | |||||
| var resourcePools modelarts.ResourcePool | var resourcePools modelarts.ResourcePool | ||||
| if err := json.Unmarshal([]byte(setting.ResourcePools), &resourcePools); err != nil { | if err := json.Unmarshal([]byte(setting.ResourcePools), &resourcePools); err != nil { | ||||
| ctx.ServerError("json.Unmarshal failed:", err) | ctx.ServerError("json.Unmarshal failed:", err) | ||||