Downloads · 30 days
117
16% of all-time downloads
KD-gpt2-120M
MiniLLM/KD-gpt2-120M
KD-gpt2-120M is a text generation model from MiniLLM. Use it when you need the model to write or continue text. The card lists the license as apache-2.0.
All-time downloads
732
Public
Repo size
996 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin498 MB · 99%
At a glance
- Task
- Text Generation
- License
- apache-2.0
- Model type
- gpt2
- Access
- Public
- Created
- Sep 26, 2024
- Updated
- Sep 26, 2024
- SHA
- 663364f9
Try a prompt
Base models
Datasets
Model card
From the Hugging Face model README
KD-gpt2-120M
KD-gpt2-120M is a gpt2-base (120M) model distilled from gpt2-xlarge (1.5B) on databricks-dolly-15k with token-level forward KLD.
It is used as a baseline for MiniLLM.
Other Baselines
Citation
@inproceedings{minillm,
title={MiniLLM: Knowledge Distillation of Large Language Models},
author={Gu, Yuxian and Dong, Li and Wei, Furu and Huang, Minlie},
booktitle={Proceedings of ICLR},
year={2024}
}
- Task
- Text Generation
- Type
- gpt2
- License
- apache-2.0
- Languages
- en
- Created
- Sep 26, 2024
- Updated
- Sep 26, 2024