SQLGenerator / pytorch_model.bin
Kasivs's picture
Create pytorch_model.bin
6d8bcb8 verified
raw
history blame
141 Bytes
import torch
# Assuming `model` is your trained model
model_save_path = "pytorch_model.bin"
torch.save(model.state_dict(), model_save_path)