CyanMonkey commited on
Commit
e28a708
·
0 Parent(s):

Duplicate from CyanMonkey/Exp-4

Browse files
.gitattributes ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ tags:
4
+ - causal-lm
5
+ - gqa
6
+ - rope
7
+ - swiglu
8
+ license: apache-2.0
9
+ ---
10
+
11
+ # Exp-4
12
+
13
+ Small language model (3.0M parameters) trained from scratch.
14
+
15
+ ## Architecture
16
+ | Property | Value |
17
+ |---|---|
18
+ | Layers | 5 |
19
+ | Hidden size | 192 |
20
+ | Intermediate size | 512 |
21
+ | Attention heads | 6 (GQA kv=6) |
22
+ | Max sequence length | 1024 |
23
+ | Vocab size | 4096 |
24
+ | Tied embeddings | True |
25
+ | Total parameters | 3.000M |
26
+
27
+ ## Training
28
+ - Tokens seen: 11,313,561,600
29
+ - Val loss: 2.3259
30
+ - Val PPL: 10.24
31
+
32
+ ## Usage
33
+ ```python
34
+ from transformers import AutoTokenizer, AutoModelForCausalLM
35
+ tokenizer = AutoTokenizer.from_pretrained("CyanMonkey/Exp-4")
36
+ model = AutoModelForCausalLM.from_pretrained("CyanMonkey/Exp-4")
37
+ inputs = tokenizer("Hello", return_tensors="pt")
38
+ output = model.generate(**inputs, max_new_tokens=50)
39
+ print(tokenizer.decode(output[0], skip_special_tokens=True))
40
+ ```
config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "LlamaForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 2,
8
+ "eos_token_id": 3,
9
+ "head_dim": 32,
10
+ "hidden_act": "silu",
11
+ "hidden_size": 192,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 512,
14
+ "max_position_embeddings": 1024,
15
+ "mlp_bias": false,
16
+ "model_type": "llama",
17
+ "num_attention_heads": 6,
18
+ "num_hidden_layers": 5,
19
+ "num_key_value_heads": 6,
20
+ "pad_token_id": 0,
21
+ "pretraining_tp": 1,
22
+ "rms_norm_eps": 1e-05,
23
+ "rope_scaling": null,
24
+ "rope_theta": 2500.0,
25
+ "tie_word_embeddings": true,
26
+ "torch_dtype": "float32",
27
+ "transformers_version": "4.40.0",
28
+ "use_cache": true,
29
+ "vocab_size": 4096
30
+ }
generation_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 2,
3
+ "eos_token_id": 3
4
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb05ffee7d46bbed7892fa146c889b15b31b899b38545f69454cecc578d5ba54
3
+ size 12006544
resume/ckpt.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7af2ef3031b31c735eb887b810a500b1429de501f9190062117f5924be8f5b46
3
+ size 36728175
resume/latest_step.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ 10800
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "bos_token": "<|bos|>",
4
+ "clean_up_tokenization_spaces": false,
5
+ "eos_token": "<|eos|>",
6
+ "is_local": false,
7
+ "local_files_only": false,
8
+ "mask_token": "<|mask|>",
9
+ "model_max_length": 4096,
10
+ "pad_token": "<|pad|>",
11
+ "tokenizer_class": "TokenizersBackend",
12
+ "unk_token": "<|unk|>"
13
+ }
training_meta.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "step": 10800,
3
+ "val_loss": 2.3259349578094484,
4
+ "val_ppl": 10.236246071401576,
5
+ "params_M": 3.0,
6
+ "pushed_at": "2026-07-03T15:24:59.309933",
7
+ "tokens_seen": 11313561600
8
+ }