初始化项目,由ModelHub XC社区提供模型
Model: 42ailab/OLMo3-190M-zh Source: Original Platform
This commit is contained in:
35
.gitattributes
vendored
Normal file
35
.gitattributes
vendored
Normal file
@@ -0,0 +1,35 @@
|
|||||||
|
*.7z filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.arrow filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.bin filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.ftz filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.gz filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.h5 filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.joblib filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.model filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.npy filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.npz filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.onnx filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.ot filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.parquet filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.pb filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.pickle filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.pkl filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.pt filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.pth filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.rar filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
||||||
|
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.tar filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.tflite filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.tgz filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.wasm filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.xz filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.zip filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.zst filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
||||||
68
README.md
Normal file
68
README.md
Normal file
@@ -0,0 +1,68 @@
|
|||||||
|
---
|
||||||
|
license: apache-2.0
|
||||||
|
language:
|
||||||
|
- zh
|
||||||
|
tags:
|
||||||
|
- pretrained
|
||||||
|
- olmo3
|
||||||
|
- chinese
|
||||||
|
library_name: transformers
|
||||||
|
pipeline_tag: text-generation
|
||||||
|
---
|
||||||
|
|
||||||
|
# OLMo3-190M-zh (v3, base)
|
||||||
|
|
||||||
|
**从零训练的 190M 中文 base 模型**,OLMo3 canonical 架构。活水 42ailab 出品,配套《零基础 AI 大模型研发训练营》L04 预训练讲。
|
||||||
|
|
||||||
|
## 用法
|
||||||
|
|
||||||
|
```python
|
||||||
|
from transformers import AutoModelForCausalLM, AutoTokenizer
|
||||||
|
tok = AutoTokenizer.from_pretrained("42ailab/OLMo3-190M-zh")
|
||||||
|
model = AutoModelForCausalLM.from_pretrained("42ailab/OLMo3-190M-zh")
|
||||||
|
```
|
||||||
|
|
||||||
|
## 训练
|
||||||
|
|
||||||
|
- 架构:OLMo3-190M(d=768, L=12, ffn=3072, QK-Norm, SWA=4096)
|
||||||
|
- Tokenizer:48k 中文 BPE(自训)
|
||||||
|
- 数据:Ultra-FineWeb-zh + Fineweb-Edu-Chinese V2.1(合计 3.4B tokens)
|
||||||
|
- 训练:Modal H100,14500 steps × 262K tokens/step,cosine 5e-4 → 5e-5
|
||||||
|
- Final mean train loss: 3.953
|
||||||
|
|
||||||
|
## 7-prompt 抽测(v3 基线)
|
||||||
|
|
||||||
|
| Prompt | 结果 |
|
||||||
|
|---|---|
|
||||||
|
| 人工智能是 | 🟢 流畅科普 |
|
||||||
|
| 山里有座庙 | 🟡 主题漂移 |
|
||||||
|
| 今天天气不错,我准备 | 🟡 漂移 |
|
||||||
|
| 北京大学位于 | 🔴 "江苏省"(错)|
|
||||||
|
| 四大发明是 | 🔴 "铜管/铁器/铜锤/铜镐"(全错)|
|
||||||
|
| 《红楼梦》人物 | 🟡 对错混杂 |
|
||||||
|
| Python 是一种 | 🔴 "开源库"(错)|
|
||||||
|
|
||||||
|
**合计 1 绿 / 3 黄 / 3 红** — 这是 base 的状态。改进版见 [`42ailab/OLMo3-190M-zh-v3.1`](https://huggingface.co/42ailab/OLMo3-190M-zh-v3.1)(continue pretrain 后"北大在北京"/"Python 是编程语言"等改善)。
|
||||||
|
|
||||||
|
## 关系
|
||||||
|
|
||||||
|
```
|
||||||
|
v3 (本 repo) ──continue pretrain──▶ v3.1 (42ailab/OLMo3-190M-zh-v3.1)
|
||||||
|
```
|
||||||
|
|
||||||
|
## License
|
||||||
|
|
||||||
|
- 权重 Apache-2.0
|
||||||
|
- 训练数据主要 Apache-2.0;不含 Wikipedia(v3.1 才引入 wiki)
|
||||||
|
|
||||||
|
## Citation
|
||||||
|
|
||||||
|
```bibtex
|
||||||
|
@misc{huoshui-olmo3-190m-zh,
|
||||||
|
title={OLMo3-190M-zh: Chinese Pretrain Teaching Model},
|
||||||
|
author={活水 AI 实验室 (42ailab) and 阳志平},
|
||||||
|
year={2026},
|
||||||
|
howpublished={\url{https://huggingface.co/42ailab/OLMo3-190M-zh}},
|
||||||
|
note={LLM001 Course, Lecture 04}
|
||||||
|
}
|
||||||
|
```
|
||||||
45
config.json
Normal file
45
config.json
Normal file
@@ -0,0 +1,45 @@
|
|||||||
|
{
|
||||||
|
"architectures": [
|
||||||
|
"Olmo3ForCausalLM"
|
||||||
|
],
|
||||||
|
"attention_bias": false,
|
||||||
|
"attention_dropout": 0.0,
|
||||||
|
"attn_implementation": "flash_attention_2",
|
||||||
|
"bos_token_id": 2,
|
||||||
|
"dtype": "bfloat16",
|
||||||
|
"eos_token_id": 0,
|
||||||
|
"hidden_act": "silu",
|
||||||
|
"hidden_size": 768,
|
||||||
|
"initializer_range": 0.02,
|
||||||
|
"intermediate_size": 3072,
|
||||||
|
"layer_types": [
|
||||||
|
"sliding_attention",
|
||||||
|
"sliding_attention",
|
||||||
|
"sliding_attention",
|
||||||
|
"full_attention",
|
||||||
|
"sliding_attention",
|
||||||
|
"sliding_attention",
|
||||||
|
"sliding_attention",
|
||||||
|
"full_attention",
|
||||||
|
"sliding_attention",
|
||||||
|
"sliding_attention",
|
||||||
|
"sliding_attention",
|
||||||
|
"full_attention"
|
||||||
|
],
|
||||||
|
"max_position_embeddings": 4096,
|
||||||
|
"model_type": "olmo3",
|
||||||
|
"num_attention_heads": 12,
|
||||||
|
"num_hidden_layers": 12,
|
||||||
|
"num_key_value_heads": 12,
|
||||||
|
"pad_token_id": 1,
|
||||||
|
"rms_norm_eps": 1e-06,
|
||||||
|
"rope_parameters": {
|
||||||
|
"rope_theta": 500000,
|
||||||
|
"rope_type": "default"
|
||||||
|
},
|
||||||
|
"sliding_window": 4096,
|
||||||
|
"tie_word_embeddings": false,
|
||||||
|
"transformers_version": "5.5.4",
|
||||||
|
"use_cache": false,
|
||||||
|
"vocab_size": 48000
|
||||||
|
}
|
||||||
12
generation_config.json
Normal file
12
generation_config.json
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
{
|
||||||
|
"_from_model_config": true,
|
||||||
|
"bos_token_id": 2,
|
||||||
|
"eos_token_id": [
|
||||||
|
0
|
||||||
|
],
|
||||||
|
"output_attentions": false,
|
||||||
|
"output_hidden_states": false,
|
||||||
|
"pad_token_id": 1,
|
||||||
|
"transformers_version": "5.5.4",
|
||||||
|
"use_cache": true
|
||||||
|
}
|
||||||
3
model.safetensors
Normal file
3
model.safetensors
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
version https://git-lfs.github.com/spec/v1
|
||||||
|
oid sha256:4c864b17cef446d37b1cb816f9e509da9d856fab45913a7dcd949bab8d2c0ced
|
||||||
|
size 374038864
|
||||||
6
special_tokens_map.json
Normal file
6
special_tokens_map.json
Normal file
@@ -0,0 +1,6 @@
|
|||||||
|
{
|
||||||
|
"bos_token": "<|bos|>",
|
||||||
|
"eos_token": "<|endoftext|>",
|
||||||
|
"pad_token": "<|pad|>",
|
||||||
|
"unk_token": "<|unk|>"
|
||||||
|
}
|
||||||
239058
tokenizer.json
Normal file
239058
tokenizer.json
Normal file
File diff suppressed because it is too large
Load Diff
12
tokenizer_config.json
Normal file
12
tokenizer_config.json
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
{
|
||||||
|
"tokenizer_class": "PreTrainedTokenizerFast",
|
||||||
|
"bos_token": "<|bos|>",
|
||||||
|
"eos_token": "<|endoftext|>",
|
||||||
|
"pad_token": "<|pad|>",
|
||||||
|
"unk_token": "<|unk|>",
|
||||||
|
"add_bos_token": false,
|
||||||
|
"add_eos_token": false,
|
||||||
|
"clean_up_tokenization_spaces": false,
|
||||||
|
"model_max_length": 4096,
|
||||||
|
"chat_template": "{% for message in messages %}{% if message['role'] == 'system' %}<|im_start|>system\n{{ message['content'] }}<|im_end|>\n{% elif message['role'] == 'user' %}<|im_start|>user\n{{ message['content'] }}<|im_end|>\n{% elif message['role'] == 'assistant' %}{% generation %}<|im_start|>assistant\n{{ message['content'] }}<|im_end|>\n{% endgeneration %}{% endif %}{% endfor %}{% if add_generation_prompt %}<|im_start|>assistant\n{% endif %}\n"
|
||||||
|
}
|
||||||
3
training_args.bin
Normal file
3
training_args.bin
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
version https://git-lfs.github.com/spec/v1
|
||||||
|
oid sha256:2392df807dcd225c654aee10e228c017fd5b946cf2f6ac9e5781305f60489f49
|
||||||
|
size 4792
|
||||||
Reference in New Issue
Block a user