SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Correct quickstart example when using the past

Browse Source
This commit is contained in:
Lysandre
2020-02-10 11:25:56 -05:00
parent 63a5399bc4
commit fd639e5be3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/quickstart.md
View File

@@ -209,7 +209,7 @@ past = None
for i in range(100): for i in range(100):
print(i) print(i)
output, past = model(context, past=past) output, past = model(context, past=past)
token = torch.argmax(output[0, :]) token = torch.argmax(output[..., -1, :])
generated += [token.tolist()] generated += [token.tolist()]
context = token.unsqueeze(0) context = token.unsqueeze(0)