mertmirac commited on
Commit
3c9049c
·
verified ·
1 Parent(s): 3b8aa15

Upload 7 files

Browse files
README.md CHANGED
@@ -1,3 +1,58 @@
1
  ---
2
- license: mit
 
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ language:
3
+ - en
4
+ license: apache-2.0
5
+ tags:
6
+ - spm
7
+ - foundation-model
8
+ - causal-lm
9
+ - from-scratch
10
+ model-index:
11
+ - name: SPM-50M-Alpha
12
+ results: []
13
  ---
14
+
15
+ # SPM-50M-Alpha
16
+
17
+ **SPM-50M-Alpha** is the first model in the SPM (Strategic Pretrained Model) family.
18
+ A ~71M parameter English foundation model trained from scratch.
19
+
20
+ ## Model Architecture
21
+
22
+ | Parameter | Value |
23
+ |-----------|-------|
24
+ | Architecture | Decoder-only Transformer (Llama-style) |
25
+ | Parameters | ~71M |
26
+ | Hidden Size | 512 |
27
+ | Layers | 12 |
28
+ | Attention Heads | 8 |
29
+ | KV Heads | 4 |
30
+ | Vocab Size | 32000 |
31
+ | Max Sequence Length | 2048 |
32
+ | Positional Encoding | RoPE |
33
+ | Normalization | RMSNorm |
34
+ | Activation | SwiGLU |
35
+
36
+ ## Training
37
+
38
+ - **Dataset**: FineWeb (sample-10BT)
39
+ - **Tokenizer**: SentencePiece BPE (vocab=32000)
40
+ - **Precision**: BF16
41
+ - **Effective Batch Size**: 128
42
+
43
+ ## Usage
44
+
45
+ ```python
46
+ from transformers import LlamaTokenizer, LlamaForCausalLM
47
+
48
+ model = LlamaForCausalLM.from_pretrained("SPM-50M-Alpha")
49
+ tokenizer = LlamaTokenizer.from_pretrained("SPM-50M-Alpha")
50
+
51
+ inputs = tokenizer("The meaning of life is", return_tensors="pt")
52
+ outputs = model.generate(**inputs, max_new_tokens=50)
53
+ print(tokenizer.decode(outputs[0]))
54
+ ```
55
+
56
+ ## License
57
+
58
+ Apache 2.0
config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "SPM-50M-Alpha",
3
+ "architectures": [
4
+ "LlamaForCausalLM"
5
+ ],
6
+ "model_type": "llama",
7
+ "bos_token_id": 2,
8
+ "eos_token_id": 3,
9
+ "pad_token_id": 0,
10
+ "hidden_size": 512,
11
+ "intermediate_size": 1536,
12
+ "num_hidden_layers": 12,
13
+ "num_attention_heads": 8,
14
+ "num_key_value_heads": 4,
15
+ "max_position_embeddings": 2048,
16
+ "vocab_size": 32000,
17
+ "rms_norm_eps": 1e-05,
18
+ "rope_theta": 10000.0,
19
+ "hidden_act": "silu",
20
+ "attention_bias": false,
21
+ "attention_dropout": 0.0,
22
+ "initializer_range": 0.02,
23
+ "tie_word_embeddings": false,
24
+ "torch_dtype": "bfloat16",
25
+ "transformers_version": "4.40.0",
26
+ "spm_version": "0.1.0"
27
+ }
generation_config.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 2,
4
+ "eos_token_id": 3,
5
+ "pad_token_id": 0,
6
+ "max_new_tokens": 512,
7
+ "do_sample": true,
8
+ "temperature": 0.8,
9
+ "top_p": 0.9,
10
+ "repetition_penalty": 1.1
11
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0861bd629d000386c3cf0db0462c402bc455d05c5cde60f1ce82d88fbf4fd63e
3
+ size 282132464
special_tokens_map.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "single_word": false,
5
+ "lstrip": false,
6
+ "rstrip": false,
7
+ "normalized": true
8
+ },
9
+ "eos_token": {
10
+ "content": "</s>",
11
+ "single_word": false,
12
+ "lstrip": false,
13
+ "rstrip": false,
14
+ "normalized": true
15
+ },
16
+ "unk_token": {
17
+ "content": "<unk>",
18
+ "single_word": false,
19
+ "lstrip": false,
20
+ "rstrip": false,
21
+ "normalized": true
22
+ },
23
+ "pad_token": {
24
+ "content": "<pad>",
25
+ "single_word": false,
26
+ "lstrip": false,
27
+ "rstrip": false,
28
+ "normalized": true
29
+ }
30
+ }
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce5baec7c99a91c46c3a1cd03e5d92457a9c1aefdef12ed34bcb51e6cc934c18
3
+ size 794363
tokenizer_config.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<s>",
3
+ "eos_token": "</s>",
4
+ "unk_token": "<unk>",
5
+ "pad_token": "<pad>",
6
+ "model_max_length": 2048,
7
+ "tokenizer_class": "LlamaTokenizer",
8
+ "legacy": false,
9
+ "sp_model_kwargs": {},
10
+ "add_bos_token": true,
11
+ "add_eos_token": false
12
+ }