WaveCut's picture
Update README.md (#2)
abd9371 verified
|
raw
history blame
5.97 kB
metadata
library_name: transformers
model_name: Vikhr-Gemma-2B-instruct
base_model:
  - google/gemma-2-2b-it
language:
  - ru
license: apache-2.0
datasets:
  - Vikhrmodels/GrandMaster-PRO-MAX

💨 Vikhr-Gemma-2B-instruct

RU

Мощная инструктивная модель на основе Gemma 2 2B, обученная на русскоязычном датасете GrandMaster-PRO-MAX.

EN

A powerful instructive model based on Gemma 2 2B, trained on the Russian-language dataset GrandMaster-PRO-MAX.

Особенности:

Попробовать / Try now:

Open In Colab

Описание:

RU

Vikhr-Gemma-2B-instruct — это мощная и компактная языковая модель, обученная на датасете GrandMaster-PRO-MAX, специально доученная для обработки русского языка.

EN

Vikhr-Gemma-2B-instruct is a powerful and compact language model trained on the GrandMaster-PRO-MAX dataset, specifically designed for processing the Russian language.

Пример кода для запуска / Sample code to run:

from transformers import AutoModelForCausalLM, AutoTokenizer

# Загрузка модели и токенизатора
model_name = "Vikhrmodels/Vikhr-Gemma-2B-instruct"
model = AutoModelForCausalLM.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

# Подготовка входного текста
input_text = "Напиши стихотворение о весне в России."

# Токенизация и генерация текста
input_ids = tokenizer.encode(input_text, return_tensors="pt")
output = model.generate(input_ids, max_length=200, num_return_sequences=1, no_repeat_ngram_size=2)

# Декодирование и вывод результата
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
print(generated_text)

Ответ модели / Model response:

Весна в России – это время обновления природы, когда природа пробуждается >от зимнего сна. Вот стихотворение, отражающее эту красоту:


Весна в России

Зимняя тишина утихла, Весна в России пришла. Солнце светит, словно в сказке, В небесах – птицы в полете.

Снег пошел, ушел вдаль, И в каждом уголке – весна. Лед промерз, вода в реке – Ветры вьют, и листья поют.

Цветы распустились, как будто В честь весны, в честь жизни. Зеленая трава, как полотно, Под ногами – мягкость.

Весна в России – это чудо, Счастье, что в сердце живет. И каждый день – праздник, Когда природа в цвету.


Надеюсь, это стихотворение передало дух и красоту весны в России.

Метрики на ru_arena_general / Metrics on ru_arena_general

Model Score 95% CI Avg Tokens Std Tokens LC Score
suzume-llama-3-8B-multilingual-orpo-borda-half 90.89 +1.1 / -1.1 2495.38 1211.62 55.86
mistral-nemo-instruct-2407 50.53 +2.5 / -2.2 403.17 321.53 50.08
sfr-iterative-dpo-llama-3-8b-r 50.06 +2.1 / -2.1 516.74 316.84 50.01
gpt-3.5-turbo-0125 50.00 +0.0 / -0.0 220.83 170.30 50.00
glm-4-9b-chat 49.75 +1.9 / -2.3 568.81 448.76 49.96
c4ai-command-r-v01 48.95 +2.6 / -1.7 529.34 368.98 49.85
llama-3-instruct-8b-sppo-iter3 47.45 +2.0 / -2.2 502.27 304.27 49.63
Vikhrmodels-vikhr-gemma-2b-it 45.82 +2.4 / -2.0 722.83 710.71 49.40
suzume-llama-3-8b-multilingual 45.71 +2.4 / -1.7 641.18 858.96 49.38
yandex_gpt_pro 45.11 +2.2 / -2.5 345.30 277.64 49.30
hermes-2-theta-llama-3-8b 44.07 +2.0 / -2.2 485.99 390.85 49.15
gpt-3.5-turbo-1106 41.48 +1.9 / -2.0 191.19 177.31 48.77
llama-3-smaug-8b 40.80 +2.1 / -1.6 524.02 480.56 48.68
llama-3-8b-saiga-suzume-ties 39.94 +2.0 / -1.7 763.27 699.39 48.55
@article{nikolich2024vikhr,
  title={Vikhr: The Family of Open-Source Instruction-Tuned Large Language Models for Russian},
  author={Aleksandr Nikolich and Konstantin Korolev and Sergey Bratchikov and Nikolay Kompanets and Artem Shelmanov},
  journal={arXiv preprint arXiv:2405.13929},
  year={2024},
  url={https://arxiv.org/pdf/2405.13929}
}