Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2from peft import PeftModel
3
4base_model = "FacebookAI/xlm-roberta-base"
5repo = "kanika103/xlm-roberta-multilingual-sentiment"
6
7tokenizer = AutoTokenizer.from_pretrained(repo)
8model = AutoModelForSequenceClassification.from_pretrained(base_model, num_labels=3)
9model = PeftModel.from_pretrained(model, repo)