CastIronMind commited on
Commit
7ee55f1
·
verified ·
1 Parent(s): 7ee2e29

Initial release: model, tokenizer, card

Browse files
README.md ADDED
@@ -0,0 +1,173 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ - ru
5
+ license: apache-2.0
6
+ library_name: transformers
7
+ pipeline_tag: text-generation
8
+ pretty_name: LissaNano
9
+ tags:
10
+ - text-generation
11
+ - llama
12
+ - my-little-pony
13
+ - mlp
14
+ - roleplay
15
+ - rp
16
+ - character
17
+ - fanfiction
18
+ - small-language-model
19
+ - tiny
20
+ - edge-deployment
21
+ - from-scratch
22
+ - lissa
23
+ widget:
24
+ - text: "<system>You are Twilight Sparkle, a studious purple unicorn who loves books.<user>Twilight, I found a strange book in the library.<assistant>"
25
+ example_title: Twilight Sparkle
26
+ - text: "<system>You are Rainbow Dash, a brash and boastful pegasus who loves speed.<user>Rainbow, are you scared of thunderstorms?<assistant>"
27
+ example_title: Rainbow Dash
28
+ - text: "<system>You are Fluttershy, a shy pegasus who cares for animals.<user>Fluttershy, a bird hurt its wing in my garden.<assistant>"
29
+ example_title: Fluttershy
30
+ model_card_authors:
31
+ - CastIronMind
32
+ model-index:
33
+ - name: LissaNano
34
+ results:
35
+ - task:
36
+ type: text-generation
37
+ dataset:
38
+ name: Pony domain eval (Fimfarchive, frozen, ~10M tokens)
39
+ type: text
40
+ metrics:
41
+ - name: Perplexity
42
+ type: perplexity
43
+ value: 31.88
44
+ - name: Bits per character
45
+ type: bpc
46
+ value: 1.2091
47
+ ---
48
+
49
+ <div align="center">
50
+
51
+ # 🦄 LissaNano
52
+
53
+ ![License](https://img.shields.io/badge/license-Apache%202.0-blue.svg)
54
+ ![Model Size](https://img.shields.io/badge/parameters-32M-green.svg)
55
+ ![Architecture](https://img.shields.io/badge/architecture-Llama%20from%20scratch-orange.svg)
56
+ ![Context](https://img.shields.io/badge/context-1024%20tokens-purple.svg)
57
+ ![Vocab](https://img.shields.io/badge/vocab-16384-blue.svg)
58
+ ![PPL](https://img.shields.io/badge/PPL-31.88-brightgreen.svg)
59
+ ![Size on disk](https://img.shields.io/badge/weights-64%20MB-lightgrey.svg)
60
+
61
+ [![Hugging Face](https://img.shields.io/badge/%F0%9F%A4%97-CastIronMind-yellow.svg)](https://huggingface.co/CastIronMind)
62
+
63
+ </div>
64
+
65
+ > ✨ **Самая маленькая RP-модель по My Little Pony, которую мы смогли найти.** 32 миллиона параметров. 64 мегабайта весов. Запускается на телефоне, на Raspberry Pi и на ноутбуке без видеокарты.
66
+
67
+ > 🧠 **Обучена с нуля, а не урезана из чужой.** Архитектура Llama, словарь 16 384 токена, контекст 1024. Ни одного заимствованного веса: сначала предобучение на фандомном корпусе, потом дообучение на ролёвых диалогах.
68
+
69
+ ---
70
+
71
+ ## Что это
72
+
73
+ **LissaNano** — младшая модель семейства [Lissa](https://huggingface.co/CastIronMind). Она умеет одно, зато честно: держать персонажа и отвечать в его голосе.
74
+
75
+ Это не ассистент. Она не ответит на вопрос про капитал Франции и не напишет код. Она нужна для другого — чтобы Twilight Sparkle снова заговорила как Twilight Sparkle.
76
+
77
+ **Что она делает хорошо:**
78
+ - держит заданную персону и узнаваемый голос персонажа
79
+ - знает лор: Понивилль, Эквестрия, Пегас, Селестия, кьютимарки
80
+ - пишет от первого лица, с действиями в `*звёздочках*`
81
+ - работает там, где большие модели не запустятся вообще
82
+
83
+ ---
84
+
85
+ ## Семейство Lissa
86
+
87
+ | модель | параметров | весов | для чего |
88
+ |---|---:|---:|---|
89
+ | **LissaNano** | 32M | 64 МБ | телефон, микрокомпьютер, игрушки, встраивание |
90
+ | [LissaBrat](https://huggingface.co/CastIronMind/LissaBrat) | 103M | 206 МБ | ноутбук, лучший связный текст |
91
+ | [Lissa 1.5B](https://huggingface.co/CastIronMind/Lissa-1.5B-Instruct) | 1.45B | 2.9 ГБ | полноценный ролёвый партнёр |
92
+
93
+ ---
94
+
95
+ ## Быстрый старт
96
+
97
+ ```python
98
+ import torch
99
+ from transformers import AutoTokenizer, AutoModelForCausalLM
100
+
101
+ model_id = "CastIronMind/LissaNano"
102
+ tok = AutoTokenizer.from_pretrained(model_id)
103
+ model = AutoModelForCausalLM.from_pretrained(model_id, dtype=torch.bfloat16)
104
+ model.eval()
105
+
106
+ SYS, USR, AST, EOS = 2, 3, 4, 1
107
+
108
+ system = "You are Twilight Sparkle, a studious purple unicorn who loves books."
109
+ user = "Twilight, I found a strange book in the library."
110
+
111
+ ids = [SYS] + tok.encode(system, add_special_tokens=False) \
112
+ + [USR] + tok.encode(user, add_special_tokens=False) + [AST]
113
+ x = torch.tensor([ids], dtype=torch.long)
114
+
115
+ out = model.generate(x, max_new_tokens=120, do_sample=True,
116
+ temperature=0.8, top_p=0.9, repetition_penalty=1.1,
117
+ pad_token_id=0, eos_token_id=EOS)
118
+ print(tok.decode(out[0][len(ids):], skip_special_tokens=True))
119
+ ```
120
+
121
+ **Формат промпта.** Ровно такой, на котором модель училась:
122
+
123
+ ```
124
+ <system>{персона}<user>{реплика}<assistant>{ответ}<eos>
125
+ ```
126
+
127
+ Спецтокены: `0 <pad>`, `1 <eos>`, `2 <system>`, `3 <user>`, `4 <assistant>`.
128
+
129
+ **Настройки генерации.** `temperature 0.8`, `top_p 0.9`, `repetition_penalty 1.1`. Температуру выше 1.0 лучше не поднимать — модель маленькая и начинает петлять.
130
+
131
+ ---
132
+
133
+ ## Данные
134
+
135
+ Предобучение: фандомная проза и форумы по My Little Pony — около **1 млрд уникальных токенов** (2 эпохи).
136
+
137
+ Дообучение: **104 732** диалога из общего микса, затем **84 681** диалог только из ролёвых и персональных подмножеств — две эпохи на финальной стадии.
138
+
139
+ Оценка качества: фиксированный набор из 313 текстов Fimfarchive, ~10 млн токенов, не пересекается с обучением.
140
+
141
+ | метрика | значение |
142
+ |---|---:|
143
+ | Perplexity | **31.88** |
144
+ | Bits per character | **1.2091** |
145
+
146
+ ---
147
+
148
+ ## Железо и цена
149
+
150
+ | | |
151
+ |---|---|
152
+ | Обучение | одна потребительская видеокарта |
153
+ | Автор | один человек |
154
+ | Лицензия | Apache 2.0 |
155
+ | Вес | 64 МБ в bf16 |
156
+
157
+ Никаких кластеров, никаких спонсоров. Модель собрана целиком и полностью руками.
158
+
159
+ ---
160
+
161
+ ## Для чего использовать
162
+
163
+ - ролёвые боты и персонажи в чатах
164
+ - встраиваемые приложения: игры, визуальные новеллы, тамагочи
165
+ - эксперименты с маленькими языковыми моделями
166
+ - обучение: живой пример полного цикла «с нуля до чат-модели» на одном GPU
167
+ - дальнейший файнтюн под своего персонажа — 32M дообучается за минуты
168
+
169
+ ---
170
+
171
+ ## Лицензия
172
+
173
+ Apache 2.0. Берите, дообучайте, квантуйте, встраивайте, публикуйте свои версии.
config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "LlamaForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 1,
8
+ "dtype": "bfloat16",
9
+ "eos_token_id": 1,
10
+ "head_dim": 32,
11
+ "hidden_act": "silu",
12
+ "hidden_size": 512,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 1536,
15
+ "max_position_embeddings": 1024,
16
+ "mlp_bias": false,
17
+ "model_type": "llama",
18
+ "num_attention_heads": 16,
19
+ "num_hidden_layers": 8,
20
+ "num_key_value_heads": 2,
21
+ "pad_token_id": 0,
22
+ "pretraining_tp": 1,
23
+ "rms_norm_eps": 1e-06,
24
+ "rope_parameters": {
25
+ "rope_theta": 10000.0,
26
+ "rope_type": "default"
27
+ },
28
+ "tie_word_embeddings": true,
29
+ "transformers_version": "5.16.1",
30
+ "use_cache": false,
31
+ "vocab_size": 16384
32
+ }
generation_config.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 1,
4
+ "eos_token_id": 1,
5
+ "pad_token_id": 0,
6
+ "transformers_version": "5.16.1",
7
+ "use_cache": true
8
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f78f3ba85928dc73affe781c65a9b0c792efd4531661251c807c01b147c5bfb9
3
+ size 63988672
special_tokens_map.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "pad_token": "<pad>",
3
+ "eos_token": "<eos>",
4
+ "bos_token": "<eos>"
5
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tokenizer_class": "PreTrainedTokenizerFast",
3
+ "model_max_length": 1024,
4
+ "padding_side": "right",
5
+ "truncation_side": "right",
6
+ "pad_token": "<pad>",
7
+ "eos_token": "<eos>",
8
+ "bos_token": "<eos>",
9
+ "clean_up_tokenization_spaces": false,
10
+ "added_tokens_decoder": {
11
+ "0": {
12
+ "content": "<pad>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "1": {
20
+ "content": "<eos>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "2": {
28
+ "content": "<system>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "3": {
36
+ "content": "<user>",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ },
43
+ "4": {
44
+ "content": "<assistant>",
45
+ "lstrip": false,
46
+ "normalized": false,
47
+ "rstrip": false,
48
+ "single_word": false,
49
+ "special": true
50
+ }
51
+ }
52
+ }