File size: 325 Bytes
cf65aec | 1 2 3 4 5 6 7 8 9 10 11 12 | 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")
|