Skip to content

Commit 63b54c8

Browse files
authored
model-conversion : make causal-verify-logits fails with model names containing "." (#16215)
Signed-off-by: Jie Fu <[email protected]>
1 parent 152729f commit 63b54c8

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

examples/model-conversion/scripts/causal/compare-logits.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def main():
4848
print(f"Error: Model file not found: {model_path}")
4949
sys.exit(1)
5050

51-
model_name = os.path.splitext(os.path.basename(model_path))[0]
51+
model_name = os.path.basename(model_path)
5252
data_dir = Path("data")
5353

5454
pytorch_file = data_dir / f"pytorch-{model_name}.bin"

examples/model-conversion/scripts/utils/check-nmse.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def main():
6767
parser.add_argument('-m', '--model-path', required=True, help='Path to the model directory')
6868
args = parser.parse_args()
6969

70-
model_name = os.path.splitext(os.path.basename(args.model_path))[0]
70+
model_name = os.path.basename(args.model_path)
7171
data_dir = Path("data")
7272

7373
pytorch_file = data_dir / f"pytorch-{model_name}.bin"

0 commit comments

Comments
 (0)