summerstay / sentence-completions-gpt-2

Uses gpt-2 to find all completions of a sentence over a certain probability threshold.
13 stars 0 forks source link

Next Two Tokens #2

Closed BigSalmon2 closed 4 years ago

BigSalmon2 commented 4 years ago

I have been trying to figure this out, but I have been unable to as the beginner I am. I would be so incredibly thankful if you were able to guide me as to how I could expand the output to the next two tokens.