以下用于句子补全的 gpt2 代码生成了一些好的句子,然后以重复句子的循环结束。
from transformers import GPT2LMHeadModel, GPT2Tokenizer
import torch
# Load the pre-trained model and tokenizer
model_name = 'gpt2'
model = GPT2LMHeadModel.from_pretrained(model_name)
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
# Set the model to evaluation mode
model.eval()
#
# Input sentence
input_sentence = "I want to go to the"
for i in range(200):
# Tokenize the input sentence
input_tokens = tokenizer.encode(input_sentence, return_tensors='pt')
# Generate predictions
with torch.no_grad():
outputs = model.generate(input_tokens, max_length=len(input_tokens) + 1, num_return_sequences=1)
# Decode the generated output
generated_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(generated_output)
input_sentence = generated_output
• Few Shot Learning 中 GPT2/GPT3 模型的变化
• 如何创建列表列表
• 如何使用 python 的 unittest 来 1) 从要测试的主代码生成一个 csv 文件和 2) 测试 csv 文件是否已生成
• 所以这就像一个掷骰子游戏,就像我陷入了一个循环,我不知道如何解决它
• 如何在 webstorm 中点击感叹号 + tab 时更改生成的代码?
• 没有循环的排列
• 如何停止在 angular cli 中重命名从产品构建生成的文件
• 当顺序正确时,程序应该进入下一个级别,但在我的代码中它不会进入下一个级别