File size: 446 Bytes
7f40992 | 1 2 3 4 5 6 7 8 9 10 11 12 13 | def parse_quant_size(output: str) -> float | None:
"""Extract quant size from full output (prefers quant over model size)."""
import re
# Try quant size first across all lines
m = re.search(r"quant size\s*=\s*([0-9.]+)\s*MiB", output)
if m:
return float(m.group(1))
# Fall back to model size
m = re.search(r"model size\s*=\s*([0-9.]+)\s*MiB", output)
if m:
return float(m.group(1))
return None
|