import torch
from transformers import GPT2Tokenizer, GPT2LMHeadModel

tokenizer = GPT2Tokenizer.from_pretrained("/remote-home/share/CodePretrain/mygpt2")
model = GPT2LMHeadModel.from_pretrained("/remote-home/share/CodePretrain/mygpt2")

inputs = tokenizer("[PAD]", return_tensors="pt")
print(inputs)
outputs = model(**inputs).logits[0]
outputs = torch.max(outputs, 1)
print(outputs[1])
output = tokenizer.decode(outputs[1])
print(output)

print("this is the first")
print("this is the second")