lewis
4 years ago
1 changed files with 5 additions and 3 deletions
Unified View
Diff Options
+ 5
- 3
routers/repo/cloudbrain.go
View File
| @@ -26,7 +26,8 @@ const ( | |||||
| ) | ) | ||||
| var ( | var ( | ||||
| gpuInfos *models.GpuInfos | |||||
| gpuInfos *models.GpuInfos | |||||
| categories *models.Categories | |||||
| ) | ) | ||||
| // MustEnableDataset check if repository enable internal cb | // MustEnableDataset check if repository enable internal cb | ||||
| @@ -135,8 +136,9 @@ func CloudBrainNew(ctx *context.Context) { | |||||
| ctx.Data["benchmark_path"] = cloudbrain.BenchMarkMountPath | ctx.Data["benchmark_path"] = cloudbrain.BenchMarkMountPath | ||||
| ctx.Data["is_benchmark_enabled"] = setting.IsBenchmarkEnabled | ctx.Data["is_benchmark_enabled"] = setting.IsBenchmarkEnabled | ||||
| var categories *models.Categories | |||||
| json.Unmarshal([]byte(setting.BenchmarkCategory), &categories) | |||||
| if categories == nil { | |||||
| json.Unmarshal([]byte(setting.BenchmarkCategory), &categories) | |||||
| } | |||||
| ctx.Data["benchmark_categories"] = categories.Category | ctx.Data["benchmark_categories"] = categories.Category | ||||
| if gpuInfos == nil { | if gpuInfos == nil { | ||||