From 3498a40390973688fe93b2ba4eb159899acdc550 Mon Sep 17 00:00:00 2001 From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com> Date: Fri, 3 Jul 2026 04:09:23 -0400 Subject: [PATCH] fix: add argparser and fix Path issue --- convert_to_safetensors.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/convert_to_safetensors.py b/convert_to_safetensors.py index 19d9633..59de20f 100644 --- a/convert_to_safetensors.py +++ b/convert_to_safetensors.py @@ -1,6 +1,7 @@ #!/usr/bin/env python3 """Convert torch.save quantized shards to safetensors format.""" +import argparse import gc import torch from pathlib import Path @@ -15,7 +16,7 @@ def convert_shards(model_path): print(f"Converting {len(shards)} shards to safetensors format...\n") for i, shard_path in enumerate(shards): - print(f"Converting shard {i+1}/{len(shards)}: {shard_path.name}") + print(f"Converting shard {i+1}/{len(shards)}: {Path(shard_path).name}") # Load torch.save format ckpt = torch.load(shard_path, map_location="cpu", weights_only=False) @@ -42,4 +43,7 @@ def convert_shards(model_path): if __name__ == "__main__": - convert_shards("/data/models/Ornith-1.0-35B-nf4") + parser = argparse.ArgumentParser() + parser.add_argument("--model-path", type=str, default="/data/models/Ornith-1.0-35B-nf4") + args = parser.parse_args() + convert_shards(args.model_path)