| import torch | |
| from safetensors import safe_open | |
| from safetensors.torch import save_file | |
| for d in [1, 2]: | |
| print(d) | |
| model = torch.load(f"pytorch_model-0000{d}-of-00002.bin") | |
| print("loaded", d, len(model)) | |
| save_file(model, f"pytorch_model-0000{d}-of-00002.safetensors") | |