Views
No views yet
1import torch
2from modeling_opengpt2 import OpenGPT2LMHeadModel
3from padded_encoder import Encoder
4
5path_to_backward = 'danyaljj/opengpt2_pytorch_backward'
6
7encoder = Encoder()
8model_backward = OpenGPT2LMHeadModel.from_pretrained(path_to_backward)
9
10input = "until she finally won."
11input_ids = encoder.encode(input)
12input_ids = torch.tensor([input_ids[::-1] ], dtype=torch.int)
13print(input_ids)
14
15output = model_backward.generate(input_ids)
16
17output_text = encoder.decode(output.tolist()[0][::-1])
18
19print(output_text)