Python 是一种广泛使用的编程语言，易于学习且用途广泛。它可用于为各种任务创建程序，例如数据分析、Web 开发和人工智能 (AI) 编写。在本文中，我们将向您展示可用于创建 AI 论文作者的三行 Python 代码。

我们需要先安装 PyTorch

pip install torch torchvision torchaudio

• 1

接下来，我们需要导入并安装transformers. 为此，请在命令提示符下运行此命令：

pip install transformers

• 1

加载预训练模型

pipeline首先，从导入transformers：

from transformers import pipeline

• 1

接下来，通过运行加载模型：

generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')

• 1

使用模型

最后，我们准备好使用模型了。

prompt = 'can i make money on medium as a writer?' # can be anything
res = generator(prompt, max_length=50, do_sample=True, temperature=0.9)

• 1
• 2

输出结果

res是单个字典的列表，结果存储在 key 下generated_text。

print(r