How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("zero-shot-classification", model="cointegrated/rubert-base-cased-nli-twoway")
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("cointegrated/rubert-base-cased-nli-twoway")
model = AutoModelForSequenceClassification.from_pretrained("cointegrated/rubert-base-cased-nli-twoway", device_map="auto")
Quick Links

RuBERT for NLI (natural language inference)

This is the DeepPavlov/rubert-base-cased fine-tuned to predict the logical relationship between two short texts: entailment or not entailment.

For more details, see the card for a similar model: https://huggingface.co/cointegrated/rubert-base-cased-nli-threeway

Downloads last month
136
Safetensors
Model size
0.2B params
Tensor type
I64
·
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train cointegrated/rubert-base-cased-nli-twoway