diff --git a/examples/model-conversion/scripts/causal/compare-logits.py b/examples/model-conversion/scripts/causal/compare-logits.py index fb959f0d56..afa0d5b263 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 196a6210f0..939e3153cc 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"