Views
No views yet
1from peft import PeftModel, PeftConfig
2from transformers import AutoModelForSequenceClassification, AutoTokenizer
3import torch
4
5repo_id = "Ark2016/ruBert-large-emotions-lora"
6model = PeftModel.from_pretrained(AutoModelForSequenceClassification.from_pretrained("ai-forever/ruBert-large", num_labels=28, problem_type="multi_label_classification"), repo_id)
7tokenizer = AutoTokenizer.from_pretrained(repo_id)