fix: init pricing overwrite vram to 0 (#351)

wangqianqianjun · web-flow · commit c0a35005c7bd · 2025-09-07T22:49:29.000+08:00
* support dedicated gpus * support dedicated GPU * support dedicated GPU * fix test issue * fix init pricing override vran * Revert "fix init pricing override vran" This reverts commit d0bea18. * fix init pricing override vram
diff --git a/internal/cloudprovider/pricing/pricing.go b/internal/cloudprovider/pricing/pricing.go
@@ -94,6 +94,9 @@ func SetTflopsMapAndInitGPUPricingInfo(ctx context.Context, gpuInfos *[]config.G
 		return
 	}
 	for _, gpuInfo := range *gpuInfos {
+		if tflopsMap[gpuInfo.FullModelName] != nil {
+			continue
+		}
 		completeInfo := &CompleteGPUInfo{
 			GpuInfo: &gpuInfo,
 		}

Original file line number	Diff line number	Diff line change
`@@ -94,6 +94,9 @@ func SetTflopsMapAndInitGPUPricingInfo(ctx context.Context, gpuInfos *[]config.G`
`94`	`94`	`return`
`95`	`95`	`}`
`96`	`96`	`for _, gpuInfo := range *gpuInfos {`
	`97`	`+ if tflopsMap[gpuInfo.FullModelName] != nil {`
	`98`	`+ continue`
	`99`	`+ }`
`97`	`100`	`completeInfo := &CompleteGPUInfo{`
`98`	`101`	`GpuInfo: &gpuInfo,`
`99`	`102`	`}`