Downloads · 30 days
0
MB55/rembert-qlora
rembert-qlora is a machine learning model from MB55. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
This model is a fine-tuned version of google/rembert, optimized using QLoRA for efficient binary classification of German dialogue utterances into:
Downloads · 30 days
0
Access
Public
Updated Aug 7, 2025
Repo size
35.7 MB
Likes
0
Public
Click a slice to open those files.
.json33.4 MB · 54%
From the Hugging Face model README
This model is a fine-tuned version of google/rembert, optimized using QLoRA for efficient binary classification of German dialogue utterances into:
ADVANCE: Contribution that moves the dialogue forward (e.g. confirmations, follow-ups, elaborations)NON-ADVANCE: Other utterances (e.g. vague responses, misunderstandings, irrelevant comments)from transformers import AutoModelForSequenceClassification, AutoTokenizer
model = AutoModelForSequenceClassification.from_pretrained("MB55/rembert-qlora")
tokenizer = AutoTokenizer.from_pretrained("MB55/rembert-qlora")
inputs = tokenizer("Also das habe ich jetzt verstanden.", return_tensors="pt")
outputs = model(**inputs)
predicted_class = outputs.logits.argmax(dim=1).item()