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

pipe = pipeline("fill-mask", model="ai-forever/ruBert-base")
# Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM

tokenizer = AutoTokenizer.from_pretrained("ai-forever/ruBert-base")
model = AutoModelForMaskedLM.from_pretrained("ai-forever/ruBert-base")
Quick Links

ruBert-base

The model architecture design, pretraining, and evaluation are documented in our preprint: A Family of Pretrained Transformer Language Models for Russian.

The model is pretrained by the SberDevices team.

  • Task: mask filling
  • Type: encoder
  • Tokenizer: BPE
  • Dict size: 120 138
  • Num Parameters: 178 M
  • Training Data Volume 30 GB

Authors

Cite us

@misc{zmitrovich2023family,
      title={A Family of Pretrained Transformer Language Models for Russian}, 
      author={Dmitry Zmitrovich and Alexander Abramov and Andrey Kalmykov and Maria Tikhonova and Ekaterina Taktasheva and Danil Astafurov and Mark Baushenko and Artem Snegirev and Tatiana Shavrina and Sergey Markov and Vladislav Mikhailov and Alena Fenogenova},
      year={2023},
      eprint={2309.10931},
      archivePrefix={arXiv},
      primaryClass={cs.CL}
}
Downloads last month
12,636
Inference Providers NEW
Examples
Mask token: [MASK]

Model tree for ai-forever/ruBert-base

Adapters
45 models
Finetunes
37 models

Spaces using ai-forever/ruBert-base 4

Paper for ai-forever/ruBert-base