Address code review feedback: improve robustness

- Use CustomSize for formatSize to avoid dependency on HumanSize format
- Improve regex to match only valid numbers (integer or float)
- Remove unused strings import from safetensors package

Author: hwclass

pkg/distribution/internal/gguf/create.go

@@ -65,10 +65,10 @@ func configFromFile(path string) types.Config {
 }
 
 var (
-	// spaceBeforeUnitRegex matches one or more spaces between a number/decimal and a letter (unit)
+	// spaceBeforeUnitRegex matches one or more spaces between a valid number and a letter (unit)
 	// Used to remove spaces between numbers and units (e.g., "16.78 M" -> "16.78M")
-	// Pattern: digits/decimals, then whitespace, then letters (unit)
-	spaceBeforeUnitRegex = regexp.MustCompile(`([0-9.]+)\s+([A-Za-z]+)`)
+	// Pattern: integer or decimal number, then whitespace, then letters (unit)
+	spaceBeforeUnitRegex = regexp.MustCompile(`([0-9]+(?:\.[0-9]+)?)\s+([A-Za-z]+)`)
 )
 
 // normalizeUnitString removes spaces between numbers and units for consistent formatting

pkg/distribution/internal/safetensors/metadata.go

@@ -6,7 +6,6 @@ import (
 	"fmt"
 	"io"
 	"os"
-	"strings"
 
 	"github.com/docker/go-units"
 )
@@ -193,7 +192,5 @@ func formatParameters(params int64) string {
 // formatSize converts bytes to human-readable format matching Docker's style
 // Returns format like "256MB" (decimal units, no space, matching `docker images`)
 func formatSize(bytes int64) string {
-	formatted := units.HumanSize(float64(bytes))
-	// Remove space between number and unit to match Docker format (e.g., "256 MB" -> "256MB")
-	return strings.ReplaceAll(formatted, " ", "")
+	return units.CustomSize("%.2f%s", float64(bytes), 1000.0, []string{"B", "kB", "MB", "GB", "TB", "PB", "EB"})
 }