POSNEGPRETRAINED / convert_model.py
chaosbringerc's picture
Create convert_model.py
6148107 verified
Raw
History Blame Contribute Delete
346 Bytes
import torch
from transformers import AutoModelForSequenceClassification
# Load your trained model
model = AutoModelForSequenceClassification.from_pretrained("chaosbringerc/POSNEG-Dataset-API")
# Save it in Hugging Face-compatible format
model.save_pretrained("./") # Saves pytorch_model.bin and config.json
print("Model saved successfully!")