cmedina/agenx-lora-training
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Path conversion for shard_path

Browse Source
This commit is contained in:
Christian Medina
2026-07-03 04:10:54 -04:00
parent 3498a40390
commit d2dde1079f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
convert_to_safetensors.py
View File

@@ -31,7 +31,7 @@ def convert_shards(model_path):
print(f" Skipping non-tensor: {key} ({type(value)})") print(f" Skipping non-tensor: {key} ({type(value)})")
# Save as safetensors # Save as safetensors
new_path = output_path / shard_path.name new_path = output_path / Path(shard_path).name
save_file(tensors, new_path) save_file(tensors, new_path)
print(f" ✓ Saved {len(tensors)} tensors") print(f" ✓ Saved {len(tensors)} tensors")