import torch | |
from safetensors.torch import load_file | |
from transformers import AutoConfig, AutoModel | |
# Path to your safetensors file | |
safetensors_path = "model.safetensors" | |
bin_path = "pytorch_model.bin" | |
# Load model weights from safetensors | |
state_dict = load_file(safetensors_path) | |
# Save as pytorch_model.bin | |
torch.save(state_dict, bin_path) | |
print(f"Converted {safetensors_path} to {bin_path}.") | |