From fa27bf663c253ad897a95221491c18798aa0175c Mon Sep 17 00:00:00 2001 From: Donato Capitella Date: Fri, 15 May 2026 10:21:24 +0100 Subject: [PATCH] refactor: update display name logic in benchmark script to consistently use filenames and strip multi-part suffix --- benchmark/run_mtp_bench.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/benchmark/run_mtp_bench.py b/benchmark/run_mtp_bench.py index 8217523..b33aec5 100755 --- a/benchmark/run_mtp_bench.py +++ b/benchmark/run_mtp_bench.py @@ -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,