Have you tried saving the model as a .pt file? I’ve used that method and it works.
# Saving
path = "./your_name.pt"
torch.save(model.state_dict(), path)

# Loading
model.load_state_dict(torch.load(path, map_location=torch.device("cpu")))