From 45d21502f0b67eb8a5ad244d469dcc0dfb7517a7 Mon Sep 17 00:00:00 2001 From: VictorSanh Date: Fri, 31 May 2019 01:04:16 -0400 Subject: [PATCH] update doc --- hubconfs/gpt_hubconf.py | 1 + 1 file changed, 1 insertion(+) diff --git a/hubconfs/gpt_hubconf.py b/hubconfs/gpt_hubconf.py index 763cc593e2..77162dc244 100644 --- a/hubconfs/gpt_hubconf.py +++ b/hubconfs/gpt_hubconf.py @@ -146,6 +146,7 @@ def openAIGPTLMHeadModel(*args, **kwargs): # Get the predicted last token >>> predicted_index = torch.argmax(predictions[0, -1, :]).item() >>> predicted_token = tokenizer.convert_ids_to_tokens([predicted_index])[0] + '.' """ model = OpenAIGPTLMHeadModel.from_pretrained(*args, **kwargs) return model