From 82b0a1ff66fa27654c7671b85b15ae3f5c60e34e Mon Sep 17 00:00:00 2001 From: Jie Fu Date: Wed, 24 Sep 2025 15:18:58 +0800 Subject: [PATCH] model-conversion : make causal-verify-logits fails with model names containing "." Signed-off-by: Jie Fu --- examples/model-conversion/scripts/causal/compare-logits.py | 2 +- examples/model-conversion/scripts/utils/check-nmse.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/model-conversion/scripts/causal/compare-logits.py b/examples/model-conversion/scripts/causal/compare-logits.py index fb959f0d566b4..afa0d5b263a0d 100755 --- a/examples/model-conversion/scripts/causal/compare-logits.py +++ b/examples/model-conversion/scripts/causal/compare-logits.py @@ -48,7 +48,7 @@ def main(): print(f"Error: Model file not found: {model_path}") sys.exit(1) - model_name = os.path.splitext(os.path.basename(model_path))[0] + model_name = os.path.basename(model_path) data_dir = Path("data") pytorch_file = data_dir / f"pytorch-{model_name}.bin" diff --git a/examples/model-conversion/scripts/utils/check-nmse.py b/examples/model-conversion/scripts/utils/check-nmse.py index 196a6210f048e..939e3153cc360 100755 --- a/examples/model-conversion/scripts/utils/check-nmse.py +++ b/examples/model-conversion/scripts/utils/check-nmse.py @@ -67,7 +67,7 @@ def main(): parser.add_argument('-m', '--model-path', required=True, help='Path to the model directory') args = parser.parse_args() - model_name = os.path.splitext(os.path.basename(args.model_path))[0] + model_name = os.path.basename(args.model_path) data_dir = Path("data") pytorch_file = data_dir / f"pytorch-{model_name}.bin"