truthserum / model.pt
chaosbringerc's picture
Create model.pt
cf65aec verified
Raw
History Blame Contribute Delete
325 Bytes
import torch
from transformers import AutoModelForSequenceClassification
# Load your trained POSNEG1 model
model = AutoModelForSequenceClassification.from_pretrained("chaosbringerc/POSNEG1")
# Convert to TorchScript format
scripted_model = torch.jit.script(model)
# Save safely as model.pt
scripted_model.save("model.pt")