提交代码

routers/repo/cloudbrain.go

@@ -731,6 +731,15 @@ func cloudBrainShow(ctx *context.Context, tpName base.TplName, jobType models.Jo
 					ctx.Data["resource_type"] = resourceType.Value
 				}
 			}
+			for _, specialPool := range cloudbrain.SpecialPools.Pools {
+
+				for _, resourceType := range specialPool.Pool {
+					if resourceType.Queue == jobRes.Config.GpuType {
+						ctx.Data["resource_type"] = resourceType.Value
+					}
+				}
+			}
+
 		} else if task.JobType == string(models.JobTypeInference) {
 			if inferenceGpuInfos == nil {
 				json.Unmarshal([]byte(setting.InferenceGpuTypes), &inferenceGpuInfos)