riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 18:51:14 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

update doc

Browse Source
This commit is contained in:
VictorSanh 2019-05-31 01:04:16 -04:00
parent 98f5c7864f
commit 45d21502f0
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
hubconfs/gpt_hubconf.py
View File

@ -146,6 +146,7 @@ def openAIGPTLMHeadModel(*args, **kwargs):
# Get the predicted last token # Get the predicted last token
>>> predicted_index = torch.argmax(predictions[0, -1, :]).item() >>> predicted_index = torch.argmax(predictions[0, -1, :]).item()
>>> predicted_token = tokenizer.convert_ids_to_tokens([predicted_index])[0] >>> predicted_token = tokenizer.convert_ids_to_tokens([predicted_index])[0]
'.</w>'
""" """
model = OpenAIGPTLMHeadModel.from_pretrained(*args, **kwargs) model = OpenAIGPTLMHeadModel.from_pretrained(*args, **kwargs)
return model return model