Downloads · 30 days
6
3% of all-time downloads
Ark2016/ruBert-large-emotions-lora
ruBert-large-emotions-lora is a text classification model from Ark2016. Use it when you need a label for a piece of text. It is set up for peft. The card lists the license as mit.
Model trained by Ark2016. Base model: ai-forever/ruBert-large.
Downloads · 30 days
6
3% of all-time downloads
All-time downloads
231
Public
Repo size
57 MB
Likes
1
Public
Click a slice to open those files.
.safetensors57 MB · 91%
From the Hugging Face model README
Model trained by Ark2016. Base model: ai-forever/ruBert-large.
from peft import PeftModel, PeftConfig
from transformers import AutoModelForSequenceClassification, AutoTokenizer
import torch
repo_id = "Ark2016/ruBert-large-emotions-lora"
model = PeftModel.from_pretrained(AutoModelForSequenceClassification.from_pretrained("ai-forever/ruBert-large", num_labels=28, problem_type="multi_label_classification"), repo_id)
tokenizer = AutoTokenizer.from_pretrained(repo_id)