Text Generation
PyTorch
English
opt
t1101675 commited on
Commit
740e31c
1 Parent(s): faf53b2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +34 -3
README.md CHANGED
@@ -1,3 +1,34 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ datasets:
4
+ - databricks/databricks-dolly-15k
5
+ language:
6
+ - en
7
+ metrics:
8
+ - rouge
9
+ base_model:
10
+ - facebook/opt-2.7B
11
+ pipeline_tag: text-generation
12
+ ---
13
+ # SeqKD-OPT-2.7B
14
+
15
+ [paper](https://arxiv.org/abs/2306.08543) | [code](https://github.com/microsoft/LMOps/tree/main/minillm)
16
+
17
+ **SeqKD-OPT-2.7B** is an OPT-2.7B model distilled from [OPT-13B](https://huggingface.co/MiniLLM/teacher-OPT-13B) on [databricks-dolly-15k](https://huggingface.co/datasets/aisquared/databricks-dolly-15k) with sequence-level forward KLD.
18
+
19
+ It is used as a baseline for [MiniLLM](https://huggingface.co/MiniLLM/MiniLLM-OPT-2.7B).
20
+
21
+ ## Other Baselines
22
+ + [SFT w/o KD](https://huggingface.co/MiniLLM/SFT-OPT-2.7B)
23
+ + [KD](https://huggingface.co/MiniLLM/KD-OPT-2.7B)
24
+
25
+
26
+ ## Citation
27
+ ```
28
+ @inproceedings{minillm,
29
+ title={MiniLLM: Knowledge Distillation of Large Language Models},
30
+ author={Gu, Yuxian and Dong, Li and Wei, Furu and Huang, Minlie},
31
+ booktitle={Proceedings of ICLR},
32
+ year={2024}
33
+ }
34
+ ```