NotHotTryHard commited on
Commit
8fd8453
·
verified ·
1 Parent(s): 9b5e923

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -7
README.md CHANGED
@@ -1,9 +1,30 @@
1
- ---
2
- tags:
3
- - model_hub_mixin
4
- - pytorch_model_hub_mixin
5
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
 
7
  This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
8
- - Library: [More Information Needed]
9
- - Docs: [More Information Needed]
 
1
+ # Transformer для генерации русских анекдотов
2
+
3
+ ## Описание задачи
4
+ Эта модель обучена на большом наборе русских анекдотов. Архитектура модели основана на трансформере с конфигурацией `mini` и использует:
5
+ - **Attention with Linear Biases (ALiBi)** для позиционного кодирования.
6
+
7
+ Дополнительно была реализована модификация модели с использованиием:
8
+ - **Multi-Headed Linear Attention (MHLA)** для эффективного вычисления внимания,
9
+ - **Rotary Positional Embedding (RoPE)** для позиционного кодирования.
10
+
11
+ Модель предназначена для генерации коротких юмористических текстов в стиле анекдотов.
12
+
13
+ ## Репорт качества
14
+ - **Метрики обучения:**
15
+ - Обучающий loss: *3.898*
16
+ - Валидационный loss: *3.989*
17
+ - **График обучения:**
18
+ ![График обучения](loss_curve.png)
19
+ - **Анализ:**
20
+ Модель демонстрирует стабильное снижение loss, однако примитивная архитектура модели не позволяет генерировать доконца осознанные фразы.
21
+
22
+ ## Примеры генерации
23
+ **Промт:** "Заходит в бар улитка"
24
+ **Генерация:** "Заходит в бар улитка и говорит: - Доктор. - Мужик."
25
+
26
+ **Промт:** "Штирлиц пришел домой"
27
+ **Генерация:** "Штирлиц пришел домой, что сегодня сченню. Штирлиц, неопрятный, и говорит: - Подайте...? "
28
 
29
  This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
30
+ - Library: [More Information Needed]