Maggio33 commited on
Commit
597edf6
·
verified ·
1 Parent(s): fbee5f5

GoLLeM-110M-PL-v3 (v2 +1 epoch on same 2B; honest internal eval, CBD regression noted)

Browse files
README.md ADDED
@@ -0,0 +1,95 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: cc-by-sa-4.0
3
+ language:
4
+ - pl
5
+ library_name: transformers
6
+ pipeline_tag: text-generation
7
+ base_model: Maggio33/GoLLeM-110M-PL-v2
8
+ datasets:
9
+ - SlayerLab/polish-dynaword
10
+ tags:
11
+ - gpt2
12
+ - polish
13
+ - base-model
14
+ - from-scratch
15
+ - amd-rocm
16
+ - continued-pretraining
17
+ ---
18
+
19
+ # GoLLeM-110M-PL-v3
20
+
21
+ Polski model językowy **110M** (GPT-2-class), trenowany **od zera** na AMD Radeon RX 7900 XTX (ROCm/WSL2). Model **bazowy (completion)** — kontynuuje tekst, **nie jest chatbotem** (nie odpowiada na pytania; daj mu początek zdania).
22
+
23
+ Trzecia iteracja serii GoLLeM. **Główna zmiana vs v2: druga epoka na tym samym czystym korpusie ~2,0 mld** (podwojona ekspozycja, ~18 → ~36 tokenów/parametr) — celowa korekta niedotrenowania v2.
24
+
25
+ > To model **completion**: dostaje początek tekstu i kontynuuje. **Nie jest chatbotem.**
26
+ > Dobrze: `"Stolica Polski to"` · Źle: `"Jaka jest stolica Polski?"`
27
+
28
+ ## Co nowego vs v2
29
+
30
+ - **Korekta niedotrenowania.** v2 widział korpus 1 raz (~18 tok/param — poniżej sufitu dla tej skali). v3 to **kontynuacja pretreningu przez 2. epokę** (łącznie ~4 mld tokenów widzianych z tego samego 2,0 mld korpusu).
31
+ - **Zmierzony efekt** (własny harness, OpenPL `polish4`, 0-shot, sygnałowe zadania): duży skok na sentymencie i redystrybucja per-zadanie (patrz Ewaluacja).
32
+ - **Uczciwie:** to podwojona **ekspozycja na te same dane**, nie nowe dane. Część zysku na zadaniach typu F1 to wyjście modelu z degeneracji (przewidywania jednoklasowego), nie gładki przyrost.
33
+
34
+ ## Trening
35
+
36
+ | | |
37
+ |---|---|
38
+ | Parametry | 110 025 216 (110M), weight-tied |
39
+ | Architektura | GPT-2 (decoder-only): 12 warstw / 12 głów / d_model 768 |
40
+ | Kontekst | 512 tokenów |
41
+ | Tokenizer | polski BPE (dynaword-32k), słownik 32 000, `<\|endoftext\|>`=0 |
42
+ | Dane | korpus v2 ~2,0 mld (58% curated: Wikipedia/Wikisource/Wolne Lektury/1000 Novels/Wiki\*/eltec + 42% HPLT v3 web clean; **zero legalese**), **2 epoki** (~4 mld tok widzianych) |
43
+ | Trening | kontynuacja z ckpt v2 (krok 60 733 → 121 466), bf16, AdamW, cosine LR + warmup, wd 0.1, batch 64 (grad-accum 4) |
44
+ | Sprzęt | 1× AMD Radeon RX 7900 XTX 24GB (gfx1100), ROCm/WSL2, ~9 h |
45
+
46
+ ## Ewaluacja
47
+
48
+ Protokół: OpenPL (`polish4`, SpeakLeash lm-evaluation-harness fork), **0-shot**, 9 zadań. Scoring **INTERNAL** (surowe acc/F1) — **porównuj DELTĘ v2→v3, nie absolut**: nie zawiera warstwy domain-PMI z tablicy [OrisTeam Polish-SLM-Benchmark](https://huggingface.co/spaces/OrisTeam/Polish-SLM-Benchmark), więc liczby absolutne **nie są** porównywalne z tą tablicą.
49
+
50
+ | zadanie | v2 | v3 | Δ |
51
+ |---|--:|--:|--:|
52
+ | PolEmo2-in | 26,7 | 41,7 | **+15,0** |
53
+ | PolEmo2-out | 10,5 | 36,8 | **+26,3** |
54
+ | 8tags | 12,8 | 13,4 | +0,6 |
55
+ | CBD (hate) | 10,6 | 3,5 | **−7,1** |
56
+ | KLEJ-NER | 6,9 | 6,9 | 0,0 |
57
+ | PSC | 0,0\* | 37,6 | +37,6\* |
58
+ | DYK | 0,0\* | 28,6 | +28,6\* |
59
+ | **Sygnał-6** | **11,3** | **23,3** | **+12,0** |
60
+
61
+ \* v2 był degeneratem na tych zadaniach F1 (predykcja jednoklasowa → F1=0); „przyrost" to wyjście z zera.
62
+
63
+ **Wniosek:** model bazowy był niedotrenowany; 2. epoka dała realne douczenie (PolEmo-in **i** -out rosną razem — to nie memoryzacja). **Regresja:** CBD (mowa nienawiści) spadła. **Board (OrisTeam) rescore = w toku** — absolutna pozycja na tablicy niepotwierdzona.
64
+
65
+ ## Użycie
66
+
67
+ ```python
68
+ from transformers import AutoModelForCausalLM, AutoTokenizer
69
+ import torch
70
+ m = AutoModelForCausalLM.from_pretrained("Maggio33/GoLLeM-110M-PL-v3").eval()
71
+ t = AutoTokenizer.from_pretrained("Maggio33/GoLLeM-110M-PL-v3")
72
+ ids = t("Stolica Polski to", return_tensors="pt").input_ids
73
+ ids = torch.cat([torch.tensor([[0]]), ids], 1) # BOS = <|endoftext|>=0
74
+ out = m.generate(ids, max_new_tokens=80, do_sample=True, temperature=0.7,
75
+ top_k=40, repetition_penalty=1.3, pad_token_id=0)
76
+ print(t.decode(out[0].tolist()[1:], skip_special_tokens=True))
77
+ ```
78
+
79
+ ## Ograniczenia
80
+
81
+ - **110M = mały** → konfabuluje konkretne fakty; uczy się głównie płynności i formy polskiego.
82
+ - **Base/completion, nie chat** — do rozmowy potrzebny SFT/instruct-tuning.
83
+ - Kontekst 512 tokenów. Brak filtrów bezpieczeństwa na wyjściu.
84
+ - **PII** scrubowane w treningu (telefon/e-mail/PESEL/NIP → tagi) → model zwraca tagi, nie realne dane; generowane imiona/adresy to konfabulacje.
85
+ - **CBD (mowa nienawiści) regresja vs v2** — do zastosowań wrażliwych na detekcję hate: użyj v2 lub zweryfikuj.
86
+
87
+ ## Licencja i atrybucja
88
+
89
+ **Korpus = CC-BY-SA-4.0** (dominująca, share-alike): Wikipedia/Wikisource/Wiki\* — CC-BY-SA-3.0 (Wikimedia Foundation); Wolne Lektury — CC-BY-SA-4.0 / Wolna Sztuka 1.3; 1000 Novels, eltec_pol — CC-BY-4.0; HPLT v3 (web) — CC0-1.0. Użycie wymaga **ATTRIBUTION** (Wikimedia Foundation, Wolne Lektury, autorzy 1000 Novels, HPLT/CLARIN-PL) oraz **SHARE-ALIKE**. **Model:** CC-BY-SA-4.0.
90
+
91
+ ## Reprodukcja
92
+
93
+ Trening: `train_125m.py --run-id gollem_v3_e2b --data gollem_v2_train_32k.bin --epochs 2 --batch 64 --accum-steps 4 --lr 3e-4` (resume z ckpt v2). Ewaluacja: `eval_slmbench.py` (polish4 0-shot + Sygnał-6). Pełny ślad: repo `amd-torch`.
94
+
95
+ **Autor:** Arkadiusz Słota.
config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_function": "gelu",
3
+ "add_cross_attention": false,
4
+ "architectures": [
5
+ "GPT2LMHeadModel"
6
+ ],
7
+ "attn_pdrop": 0.0,
8
+ "bos_token_id": 0,
9
+ "dtype": "float32",
10
+ "embd_pdrop": 0.0,
11
+ "eos_token_id": 0,
12
+ "initializer_range": 0.02,
13
+ "layer_norm_epsilon": 1e-05,
14
+ "model_type": "gpt2",
15
+ "n_ctx": 512,
16
+ "n_embd": 768,
17
+ "n_head": 12,
18
+ "n_inner": null,
19
+ "n_layer": 12,
20
+ "n_positions": 512,
21
+ "pad_token_id": null,
22
+ "reorder_and_upcast_attn": false,
23
+ "resid_pdrop": 0.0,
24
+ "scale_attn_by_inverse_layer_idx": false,
25
+ "scale_attn_weights": true,
26
+ "summary_activation": null,
27
+ "summary_first_dropout": 0.1,
28
+ "summary_proj_to_labels": true,
29
+ "summary_type": "cls_index",
30
+ "summary_use_proj": true,
31
+ "tie_word_embeddings": true,
32
+ "transformers_version": "5.16.1",
33
+ "use_cache": true,
34
+ "vocab_size": 32000
35
+ }
generation_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "eos_token_id": 0,
5
+ "output_attentions": false,
6
+ "output_hidden_states": false,
7
+ "transformers_version": "5.16.1",
8
+ "use_cache": true
9
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a9dc41cbfcf8327163e48e5b36ba3fe3eb4bf26aa8c9f008c7492e1f0932a097
3
+ size 440115840
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"tokenizer_class":"PreTrainedTokenizerFast","bos_token":"<|endoftext|>","eos_token":"<|endoftext|>","unk_token":"<|endoftext|>"}