#2701

update

routers/repo/cloudbrain.go

@@ -167,13 +167,8 @@ func cloudBrainNewDataPrepare(ctx *context.Context) error {
 	if cloudbrain.InferenceResourceSpecs != nil {
 		ctx.Data["inference_resource_specs"] = cloudbrain.InferenceResourceSpecs.ResourceSpec
 	}
-	specs, _ := resource.FindAvailableSpecs(ctx.User.ID, models.FindSpecsOptions{
-		JobType:         models.JobTypeDebug,
-		ComputeResource: models.GPU,
-		Cluster:         models.OpenICluster,
-		AiCenterCode:    models.AICenterOfCloudBrainOne,
-	})
-	ctx.Data["Specs"] = specs
+
+	prepareCloudbrainOneSpecs(ctx)
 
 	if cloudbrain.SpecialPools != nil {
 		var debugGpuTypes []*models.GpuInfo
@@ -226,6 +221,34 @@ func cloudBrainNewDataPrepare(ctx *context.Context) error {
 	return nil
 }
 
+func prepareCloudbrainOneSpecs(ctx *context.Context) {
+	debugSpecs, _ := resource.FindAvailableSpecs(ctx.User.ID, models.FindSpecsOptions{
+		JobType:         models.JobTypeDebug,
+		ComputeResource: models.GPU,
+		Cluster:         models.OpenICluster,
+		AiCenterCode:    models.AICenterOfCloudBrainOne,
+	})
+	ctx.Data["debug_specs"] = debugSpecs
+	b, _ := json.Marshal(debugSpecs)
+	log.Info("%s", string(b))
+
+	trainSpecs, _ := resource.FindAvailableSpecs(ctx.User.ID, models.FindSpecsOptions{
+		JobType:         models.JobTypeTrain,
+		ComputeResource: models.GPU,
+		Cluster:         models.OpenICluster,
+		AiCenterCode:    models.AICenterOfCloudBrainOne,
+	})
+	ctx.Data["train_specs"] = trainSpecs
+
+	inferenceSpecs, _ := resource.FindAvailableSpecs(ctx.User.ID, models.FindSpecsOptions{
+		JobType:         models.JobTypeInference,
+		ComputeResource: models.GPU,
+		Cluster:         models.OpenICluster,
+		AiCenterCode:    models.AICenterOfCloudBrainOne,
+	})
+	ctx.Data["inference_specs"] = inferenceSpecs
+}
+
 func CloudBrainNew(ctx *context.Context) {
 	err := cloudBrainNewDataPrepare(ctx)
 	if err != nil {