生成式模型实战—小小案例（python）

实战之前，环境需要已经搭建好。如果环境没有搭建好，可以参考PyTorch2.0 环境搭建详细步骤(Nvidia显卡)-CSDN博客

接下来，我们今天的学习之旅~

Step1. 安装类库 transformers

pip install transformers

Step2. 代码敬上

from transformers import BertTokenizer, GPT2LMHeadModel, TextGenerationPipeline

tokenizer = BertTokenizer.from_pretrained("uer/gpt2-chinese-poem")
model = GPT2LMHeadModel.from_pretrained("uer/gpt2-chinese-poem")
text_generator = TextGenerationPipeline(model, tokenizer)
result = text_generator("[CLS]空山新雨后", max_length=50, do_sample=True)
print('输出', result)

注意，跑这个的时候如果报错，可以尝试加个梯子。

如果想换成其他模型，可以 到https://huggingface.co/models

每天进步一点点~ 加油~