如何在colab中直接调用huggingface模型?
网友回复
在colab中新建笔记,输入以下内容,点击全部运行即可:
!pip install transformers
from transformers import pipeline # 使用pipeline函数下载并运行GPT-2模型 nlp = pipeline("text-generation", model="gpt2") # 使用模型生成文本 text = nlp("Hello, how are you?") print(text)