Downloads · 30 days
17
3% of all-time downloads
opengpt2_pytorch_forward
danyaljj/opengpt2_pytorch_forward
opengpt2_pytorch_forward is a machine learning model from danyaljj. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
West et al.'s model from their "reflective decoding" paper.
All-time downloads
585
Public
Repo size
11.9 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin6 GB · 100%
At a glance
- Library
- transformers
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Jun 16, 2021
- SHA
- 76cc4809
Model card
From the Hugging Face model README
West et al.'s model from their "reflective decoding" paper.
Sample usage:
import torch
from modeling_opengpt2 import OpenGPT2LMHeadModel
from padded_encoder import Encoder
path_to_forward = 'danyaljj/opengpt2_pytorch_forward'
encoder = Encoder()
model_backward = OpenGPT2LMHeadModel.from_pretrained(path_to_forward)
input = "She tried to win but"
input_ids = encoder.encode(input)
input_ids = torch.tensor([input_ids ], dtype=torch.int)
print(input_ids)
output = model_backward.generate(input_ids)
output_text = encoder.decode(output.tolist()[0])
print(output_text)
Download the additional files from here: https://github.com/peterwestuw/GPT2ForwardBackward
- Library
- transformers
- Created
- Mar 2, 2022
- Updated
- Jun 16, 2021