refactor: update display name logic in benchmark script to consistently use filenames and strip multi-part suffix

This commit is contained in:
Donato Capitella
2026-05-15 10:21:24 +01:00
parent d1e5e0b3b0
commit fa27bf663c
+4 -5
View File
@@ -96,11 +96,10 @@ def discover_models(models_dir: Path) -> list[dict]:
if "-000" in name and "-00001-of-" not in name:
continue
# Derive display name from parent directory or filename
if gguf.parent != models_dir:
display_name = gguf.parent.name
else:
display_name = gguf.stem
# Always use the actual filename (without .gguf) to capture the quant level
display_name = gguf.stem
if "-00001-of-" in display_name:
display_name = display_name.split("-00001-of-")[0]
models.append({
"name": display_name,