Add library_name, pipeline_tag and license

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +10 -10
README.md CHANGED
@@ -3,7 +3,11 @@ datasets:
3
  - HuggingFaceTB/smollm-corpus
4
  language:
5
  - en
 
 
 
6
  ---
 
7
  # Outlier-Safe Pre-Training
8
 
9
  [![arXiv](https://img.shields.io/badge/arXiv-2506.19697-b31b1b?style=flat-square)](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
- - [🤗 OSP-1.4B-1T-Adam](https://huggingface.co/dmis-lab/OSP-1.4B-1T-Adam): Trained on the standard Adam optimizer, without any modifications.
37
- - [🤗 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.
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
- &dagger;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
  [![arXiv](https://img.shields.io/badge/arXiv-2506.19697-b31b1b?style=flat-square)](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
+ &dagger;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
  ```