Views
No views yet
CPM-Generate-distill is the distill model of CPM.1from transformers import TextGenerationPipeline, AutoTokenizer, AutoModelWithLMHead
2
3tokenizer = AutoTokenizer.from_pretrained("mymusise/CPM-Generate-distill")
4model = AutoModelWithLMHead.from_pretrained("mymusise/CPM-Generate-distill")
5
6text_generater = TextGenerationPipeline(model, tokenizer)
7
8print(text_generator('清华大学是', max_length=50, do_sample=True, top_p=0.9))