vincentzhou commited on
Commit
1b1f9e8
·
verified ·
1 Parent(s): 1309cbe

llm001 L04 Full: 190M, 20-step test

Browse files
README.md ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: cmz1024/olmo3-190m-zh-full
3
+ license: apache-2.0
4
+ language:
5
+ - zh
6
+ tags:
7
+ - llm001
8
+ - olmo3
9
+ - chinese
10
+ - pretrained
11
+ ---
12
+
13
+ # OLMo3-190M-zh-full
14
+
15
+ 为零基础 AI 大模型研发训练营(llm001)L04 Full 模型(190M 参数,20 步测试训练)。
16
+
17
+ ## 模型配置
18
+
19
+ - hidden_size: 768, num_layers: 12, num_heads: 12, intermediate_size: 3072
20
+ - vocab_size: 48000, sliding_window: 4096
21
+
22
+ ## 训练配置
23
+
24
+ - 数据:cmz1024/llm101-olmo3-zh-demo-data (500M tokens)
25
+ - 训练:H100, max_steps=20, bs=16×8=128, lr=5e-4, bf16
26
+
27
+ ## 用法
28
+
29
+ ```python
30
+ from transformers import AutoModelForCausalLM, AutoTokenizer
31
+ model = AutoModelForCausalLM.from_pretrained("cmz1024/olmo3-190m-zh-full")
32
+ tok = AutoTokenizer.from_pretrained("cmz1024/olmo3-190m-zh-full")
33
+ ```
chat_template.jinja ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {% for message in messages %}{% if message['role'] == 'system' %}<|im_start|>system
2
+ {{ message['content'] }}<|im_end|>
3
+ {% elif message['role'] == 'user' %}<|im_start|>user
4
+ {{ message['content'] }}<|im_end|>
5
+ {% elif message['role'] == 'assistant' %}{% generation %}<|im_start|>assistant
6
+ {{ message['content'] }}<|im_end|>
7
+ {% endgeneration %}{% endif %}{% endfor %}{% if add_generation_prompt %}<|im_start|>assistant
8
+ {% endif %}
config.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Olmo3ForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 2,
8
+ "dtype": "bfloat16",
9
+ "eos_token_id": 0,
10
+ "hidden_act": "silu",
11
+ "hidden_size": 768,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 3072,
14
+ "layer_types": [
15
+ "sliding_attention",
16
+ "sliding_attention",
17
+ "sliding_attention",
18
+ "full_attention",
19
+ "sliding_attention",
20
+ "sliding_attention",
21
+ "sliding_attention",
22
+ "full_attention",
23
+ "sliding_attention",
24
+ "sliding_attention",
25
+ "sliding_attention",
26
+ "full_attention"
27
+ ],
28
+ "max_position_embeddings": 4096,
29
+ "model_type": "olmo3",
30
+ "num_attention_heads": 12,
31
+ "num_hidden_layers": 12,
32
+ "num_key_value_heads": 12,
33
+ "pad_token_id": 1,
34
+ "rms_norm_eps": 1e-06,
35
+ "rope_parameters": {
36
+ "rope_theta": 500000,
37
+ "rope_type": "default"
38
+ },
39
+ "sliding_window": 4096,
40
+ "tie_word_embeddings": false,
41
+ "transformers_version": "5.8.1",
42
+ "use_cache": false,
43
+ "vocab_size": 48000
44
+ }
generation_config.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 2,
4
+ "eos_token_id": [
5
+ 0
6
+ ],
7
+ "output_attentions": false,
8
+ "output_hidden_states": false,
9
+ "pad_token_id": 1,
10
+ "transformers_version": "5.8.1",
11
+ "use_cache": true
12
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:caebbb907be478e74237960d2f34bf5a4e411e567be9ee5115d5b892a278b5a6
3
+ size 374038864
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "bos_token": "<|bos|>",
4
+ "clean_up_tokenization_spaces": false,
5
+ "eos_token": "<|endoftext|>",
6
+ "is_local": true,
7
+ "local_files_only": false,
8
+ "model_max_length": 4096,
9
+ "pad_token": "<|pad|>",
10
+ "tokenizer_class": "TokenizersBackend",
11
+ "unk_token": "<|unk|>"
12
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a31e83a83d6c27e9be91eabbcca2231c6267dd1ae69e999cad191ede7e47f2f
3
+ size 4856