Downloads · 30 days
50
0% of all-time downloads
mymusise/CPM-Generate-distill
CPM-Generate-distill is a text generation model from mymusise. Use it when you need the model to write or continue text. It is set up for transformers.
<h1 align="center" CPM-Generate-distill </h1
Downloads · 30 days
50
0% of all-time downloads
All-time downloads
21.4K
Public
Repo size
2.2 GB
Likes
7
Public
Click a slice to open those files.
.bin448 MB · 34%
From the Hugging Face model README
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 The One Thing You Need to Know is this model is not uploaded by official, the conver script is here
And the CPM-Generate-distill is the distill model of CPM.
How to use this model directly from the 🤗/transformers library:
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))
