Views
No views yet
1from transformers import BertTokenizer, GPT2LMHeadModel, TextGenerationPipeline
2tokenizer = BertTokenizer.from_pretrained("zhuimengshaonian/gpt2-ancient-medium")
3model = GPT2LMHeadModel.from_pretrained("zhuimengshaonian/gpt2-ancient-medum")
4text_generator = TextGenerationPipeline(model, tokenizer)
5text_generator("[CLS]当是时", max_length=100, do_sample=True)