初始化项目,由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