Add library_name, pipeline_tag and license
#1
by nielsr HF Staff - opened
README.md
CHANGED
|
@@ -3,7 +3,11 @@ datasets:
|
|
| 3 |
- HuggingFaceTB/smollm-corpus
|
| 4 |
language:
|
| 5 |
- en
|
|
|
|
|
|
|
|
|
|
| 6 |
---
|
|
|
|
| 7 |
# Outlier-Safe Pre-Training
|
| 8 |
|
| 9 |
[](https://arxiv.org/abs/2506.19697)
|
|
@@ -25,17 +29,14 @@ A method that prevents outliers but significantly reduces efficiency is unlikely
|
|
| 25 |
3. 🧩**Ensuring full compatibility with existing inference pipelines**<br/>
|
| 26 |
We prioritize compatibility with widely adopted inference frameworks such as vLLM and SGLang. Rather than introducing architectural changes that break compatibility, OSP preserves computational invariance, allowing models to be directly integrated into existing pipelines without additional effort.
|
| 27 |
|
| 28 |
-
|
| 29 |
-
|
| 30 |
## Model Checkpoints
|
| 31 |
|
| 32 |
### Final Models
|
| 33 |
|
| 34 |
The models were trained on 1 trillion tokens, following the pre-training recipe of [SmolLM](https://huggingface.co/blog/smollm). Specifically, training was conducted using the [smollm-corpus](https://huggingface.co/datasets/HuggingFaceTB/smollm-corpus), a mixture of FineWeb-Edu, Cosmopedia, and Python-Edu.
|
| 35 |
|
| 36 |
-
-
|
| 37 |
-
-
|
| 38 |
-
|
| 39 |
|
| 40 |
### Ablation Models
|
| 41 |
|
|
@@ -175,8 +176,7 @@ The models were trained on 1 trillion tokens, following the pre-training recipe
|
|
| 175 |
</tr>
|
| 176 |
</tbody>
|
| 177 |
</table>
|
| 178 |
-
†Model configuration that disables decoupled embedding optimization by training with Muon optimizer without Adam optimization on embedding layers
|
| 179 |
-
|
| 180 |
|
| 181 |
## Training
|
| 182 |
|
|
@@ -185,7 +185,7 @@ The models were trained on 1 trillion tokens, following the pre-training recipe
|
|
| 185 |
- Architecture: Llama
|
| 186 |
- Pretraining tokens: 100 billion tokens
|
| 187 |
- Precision: bfloat16
|
| 188 |
-
|
| 189 |
### Hardware
|
| 190 |
|
| 191 |
- TPUs: TPU-v4-512 Pod Slice (supported by [TRC Program](https://sites.research.google/trc/about/))
|
|
@@ -202,12 +202,12 @@ This model family was trained to demonstrate the effectiveness of eliminating ou
|
|
| 202 |
|
| 203 |
```bibtex
|
| 204 |
@article{park2025osp,
|
| 205 |
-
title={Outlier-Safe Pre-Training for Robust 4-Bit Quantization of Large Language Models},
|
| 206 |
author={Jungwoo Park and Taewhoo Lee and Chanwoong Yoon and Hyeon Hwang and Jaewoo Kang},
|
| 207 |
year={2025},
|
| 208 |
eprint={2506.19697},
|
| 209 |
archivePrefix={arXiv},
|
| 210 |
primaryClass={cs.LG},
|
| 211 |
-
url={https://arxiv.org/abs/2506.19697},
|
| 212 |
}
|
| 213 |
```
|
|
|
|
| 3 |
- HuggingFaceTB/smollm-corpus
|
| 4 |
language:
|
| 5 |
- en
|
| 6 |
+
license: apache-2.0
|
| 7 |
+
library_name: transformers
|
| 8 |
+
pipeline_tag: text-generation
|
| 9 |
---
|
| 10 |
+
|
| 11 |
# Outlier-Safe Pre-Training
|
| 12 |
|
| 13 |
[](https://arxiv.org/abs/2506.19697)
|
|
|
|
| 29 |
3. 🧩**Ensuring full compatibility with existing inference pipelines**<br/>
|
| 30 |
We prioritize compatibility with widely adopted inference frameworks such as vLLM and SGLang. Rather than introducing architectural changes that break compatibility, OSP preserves computational invariance, allowing models to be directly integrated into existing pipelines without additional effort.
|
| 31 |
|
|
|
|
|
|
|
| 32 |
## Model Checkpoints
|
| 33 |
|
| 34 |
### Final Models
|
| 35 |
|
| 36 |
The models were trained on 1 trillion tokens, following the pre-training recipe of [SmolLM](https://huggingface.co/blog/smollm). Specifically, training was conducted using the [smollm-corpus](https://huggingface.co/datasets/HuggingFaceTB/smollm-corpus), a mixture of FineWeb-Edu, Cosmopedia, and Python-Edu.
|
| 37 |
|
| 38 |
+
- 🤗 OSP-1.4B-1T-Adam](https://huggingface.co/dmis-lab/OSP-1.4B-1T-Adam): Trained on the standard Adam optimizer, without any modifications.
|
| 39 |
+
- 🤗 OSP-1.4B-1T-Muon-SSNorm-EmbProj](https://huggingface.co/dmis-lab/OSP-1.4B-1T-Muon-SSNorm-EmbProj): Trained on the OSP framework. This is our final model.
|
|
|
|
| 40 |
|
| 41 |
### Ablation Models
|
| 42 |
|
|
|
|
| 176 |
</tr>
|
| 177 |
</tbody>
|
| 178 |
</table>
|
| 179 |
+
†Model configuration that disables decoupled embedding optimization by training with Muon optimizer without Adam optimization on embedding layers
|
|
|
|
| 180 |
|
| 181 |
## Training
|
| 182 |
|
|
|
|
| 185 |
- Architecture: Llama
|
| 186 |
- Pretraining tokens: 100 billion tokens
|
| 187 |
- Precision: bfloat16
|
| 188 |
+
|
| 189 |
### Hardware
|
| 190 |
|
| 191 |
- TPUs: TPU-v4-512 Pod Slice (supported by [TRC Program](https://sites.research.google/trc/about/))
|
|
|
|
| 202 |
|
| 203 |
```bibtex
|
| 204 |
@article{park2025osp,
|
| 205 |
+
title={Outlier-Safe Pre-Training for Robust 4-Bit Quantization of Large Language Models},
|
| 206 |
author={Jungwoo Park and Taewhoo Lee and Chanwoong Yoon and Hyeon Hwang and Jaewoo Kang},
|
| 207 |
year={2025},
|
| 208 |
eprint={2506.19697},
|
| 209 |
archivePrefix={arXiv},
|
| 210 |
primaryClass={cs.LG},
|
| 211 |
+
url={https://arxiv.org/abs/2506.19697},
|
| 212 |
}
|
| 213 |
```
|