SAVRN
Search Contact SAVRN

Open-weight model · Zero-shot classification

rubert-base-cased-nli-threeway

by David Dale cointegrated/rubert-base-cased-nli-threeway

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

Parameters178M
Context512
Weights2.1 GB
License
AccessOpen weights
Monthly Downloads18.1k

Runs On

What it takes to serve rubert-base-cased-nli-threeway (178M parameters): the memory its weights need at each precision, and the cheapest way to rent enough data-center GPUs to hold them.

PrecisionWeightsMemory neededCheapest setupPer hourAlso fits
16-bit 0.4 GB 0.4 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
8-bit 0.2 GB 0.2 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
4-bit 0.1 GB 0.1 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00

Memory is the weights at that precision plus 20% for the runtime and a short context; a long context needs more. Prices are the lowest on-demand hourly rates in the SAVRN Index, read Sep 18, 2026.

Model Card

This is the DeepPavlov/rubert-base-cased fine-tuned to predict the logical relationship between two short texts: entailment, contradiction, or neutral. How to run the model for NLI: You can also use this model for zero-shot short text classification (by labels only), e.g. for sentiment analysis: Alternatively, you can use Huggingface pipelines for inference. The model has been trained on a series of NLI datasets automatically translated to Russian from English. Most datasets were taken from the repo of Felipe Salvatore: Some datasets obtained from the original sources: The table below shows ROC AUC (one class vs rest) for five models on the corresponding dev sets: - tiny: a small BERT…

Excerpt from the card by David Dale.

Configuration

Architecture
BertForSequenceClassification
Context length (tokens)
512
Layers
12
Hidden size
768
Feed-forward size
3,072
Attention heads
12
Vocabulary size
119,547
Stored precision
float32
Model type
bert

Identity and Version

Repository
cointegrated/rubert-base-cased-nli-threeway
Publisher
David Dale
Task
Zero-shot classification
Modality
Text
Library
transformers
Parameters
178M parameters
Languages
ru
Revision
920cbb52ef830e94461bf141ec2119979b6049e2
First published
2022-03-02
Last updated
2024-04-05

Files and Weights

15 files, 2.1 GB in total. The weights are 3 files totalling 2.1 GB in bin, onnx, safetensors.

Weights3 files · 2.1 GB
Configuration4 files · 3.0 KB
Tokenizer6 files · 9.5 MB
Documentation1 file · 6.4 KB
Repository1 file · 1.2 KB
Every file
FileTypeSizeSHA-256
model.safetensorsWeights711.5 MB edde513c2192
onnx/model.onnxWeights711.7 MB 4a3bd6535220
pytorch_model.binWeights711.5 MB 7821ef90a2f2
config.jsonConfiguration1.1 KB
onnx/config.jsonConfiguration1.1 KB
onnx/special_tokens_map.jsonConfiguration695 B
special_tokens_map.jsonConfiguration112 B
README.mdDocumentation6.4 KB
.gitattributesRepository1.2 KB
onnx/tokenizer.jsonTokenizer3.6 MB
onnx/tokenizer_config.jsonTokenizer1.4 KB
onnx/vocab.txtTokenizer1.6 MB
tokenizer.jsonTokenizer2.6 MB
tokenizer_config.jsonTokenizer545 B
vocab.txtTokenizer1.6 MB

License and Download

License
Not stated by the source
Access
Open weights, no gate
Download size
2.1 GB
Download from David Dale

Released by David Dale through its official repository on Hugging Face.

Built From

  • Trained on (disclosed) cointegrated/nli-rus-translated-v2021

Memory Requirements

PrecisionWeights in memory
As published2.1 GB
16-bit0.4 GB
8-bit0.2 GB
4-bit0.1 GB

Weights only, from the published parameter count; the key-value cache and runtime add to this.

Questions About rubert-base-cased-nli-threeway

How much GPU memory does rubert-base-cased-nli-threeway need?

About 0.4 GB at 16-bit and 0.1 GB at 4-bit: the weights (178M parameters) plus a working margin. A long context needs more.

What is the cheapest GPU to run rubert-base-cased-nli-threeway on?

At 16-bit, 1x MI300X from $1.85 an hour; at 4-bit, 1x MI300X from $1.85 an hour, at the lowest on-demand prices the SAVRN Index lists.

What is rubert-base-cased-nli-threeway's context length?

512 tokens, from the maximum position embeddings in its published configuration.

Similar Models

Models in this series are designed for efficient zeroshot classification with the Hugging Face pipeline. These models can do classification without training data and run on both GPUs and CPUs. An overview of the latest zeroshot classifiers is available in my Zeroshot Classifier Collection. The main update of this zeroshot-v2.0 series of models is that several models are trained on fully commercially-friendly data for users with strict license requirements. These models can do one universal classification task: determine whether a hypothesis is "true" or "not true" given a text (entailment vs. notentailment). This task format is based on the Natural Language Inference task (NLI). The task is…

Open weights mit 184M parameters 512 tokens transformers

The model is designed for zero-shot classification with the Hugging Face pipeline. The model can do one universal classification task: determine whether a hypothesis is "true" or "not true" given a text (entailment vs. notentailment). This task format is based on the Natural Language Inference task (NLI). The task is so universal that any classification task can be reformulated into this task. A detailed description of how the model was trained and how it can be used is available in this paper. The model was trained on a mixture of 33 datasets and 387 classes that have been reformatted into this universal format. 1. Five NLI datasets with ~885k texts: "mnli", "anli", "fever", "wanli"…

Open weights mit 184M parameters 512 tokens transformers

Model · Zero-shot classification

deberta-base-long-nli

Tasksource

deberta-v3-base with context length of 1280 fine-tuned on tasksource for 250k steps. I oversampled long NLI tasks (ConTRoL, doc-nli). Training data include helpsteer v1/v2, logical reasoning tasks (FOLIO, FOL-nli, LogicNLI...), OASST, hh/rlhf, linguistics oriented NLI tasks, tasksource-dpo, fact verification tasks. This checkpoint has strong zero-shot validation performance on many tasks (e.g. 70% on WNLI), and can be used for: - Zero-shot entailment-based classification for arbitrary labels [ZS]. - Natural language inference [NLI] - Further fine-tuning on a new task or tasksource task (classification, token classification, reward modeling or multiple-choice) [FT]. Zero-shot GPT-4 scores…

Open weights apache-2.0 184M parameters 1,280 tokens transformers

This model was trained using SentenceTransformers Cross-Encoder class. This model is based on microsoft/deberta-v3-base The model was trained on the SNLI and MultiNLI datasets. For a given sentence pair, it will output three scores corresponding to the labels: contradiction, entailment, neutral. For futher evaluation results, see SBERT.net - Pretrained Cross-Encoder. Pre-trained models can be used like this: You can use the model also directly with Transformers library (without SentenceTransformers library): This model can also be used for zero-shot-classification

Open weights apache-2.0 184M parameters 512 tokens sentence-transformers

This model was trained on the MultiNLI, Fever-NLI and Adversarial-NLI (ANLI) datasets, which comprise 763 913 NLI hypothesis-premise pairs. This base model outperforms almost all large models on the ANLI benchmark. The base model is DeBERTa-v3-base from Microsoft. The v3 variant of DeBERTa substantially outperforms previous versions of the model by including a different pre-training objective, see annex 11 of the original DeBERTa paper. For highest performance (but less speed), I recommend using https://huggingface.co/MoritzLaurer/DeBERTa-v3-large-mnli-fever-anli-ling-wanli. DeBERTa-v3-base-mnli-fever-anli was trained on the MultiNLI, Fever-NLI and Adversarial-NLI (ANLI) datasets, which…

Open weights mit 184M parameters 512 tokens transformers