File size: 1,198 Bytes
35aaa80 d5291e3 35aaa80 d5291e3 35aaa80 33dadce 35aaa80 d5291e3 35aaa80 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
---
language: zh
widget:
- text: "天下熙熙，"
- text: "清华大学是"
---
<h1 align="center">
CPM-Generate-distill
</h1>
CPM(Chinese Pre-Trained Language Models), which has 2.6B parameters, made by the research team of Beijing Zhiyuan Institute of artificial intelligence and Tsinghua University @TsinghuaAI.
[repo: CPM-Generate](https://github.com/TsinghuaAI/CPM-Generate)
The One Thing You Need to Know is this model is not uploaded by official, the conver script is [here](https://github.com/mymusise/CPM-TF2Transformer/blob/main/transfor_CMP.ipynb)
And the `CPM-Generate-distill` is the distill model of `CPM`.
# How to use
How to use this model directly from the 🤗/transformers library:
```python
from transformers import TextGenerationPipeline, AutoTokenizer, AutoModelWithLMHead
tokenizer = AutoTokenizer.from_pretrained("mymusise/CPM-Generate-distill")
model = AutoModelWithLMHead.from_pretrained("mymusise/CPM-Generate-distill")
text_generater = TextGenerationPipeline(model, tokenizer)
print(text_generator('清华大学是', max_length=50, do_sample=True, top_p=0.9))
```
|