From 99d1a6944411d372a33f8a31ff8cd7a47473a86a Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Thu, 7 May 2020 10:54:48 +0200 Subject: [PATCH] fix examples (#4192) --- src/transformers/modeling_reformer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/transformers/modeling_reformer.py b/src/transformers/modeling_reformer.py index fc7aa2dc0d..dc671b0ab5 100644 --- a/src/transformers/modeling_reformer.py +++ b/src/transformers/modeling_reformer.py @@ -1515,11 +1515,11 @@ class ReformerModel(ReformerPreTrainedModel): Examples:: - from transformers import ReformerModelWithLMHead, ReformerTokenizer + from transformers import ReformerModel, ReformerTokenizer import torch tokenizer = ReformerTokenizer.from_pretrained('google/reformer-crime-and-punishment') - model = ReformerModelWithLMHead.from_pretrained('google/reformer-crime-and-punishment') + model = ReformerModel.from_pretrained('google/reformer-crime-and-punishment') input_ids = torch.tensor(tokenizer.encode("Hello, my dog is cute", add_special_tokens=True)).unsqueeze(0) # Batch size 1 outputs = model(input_ids) @@ -1716,7 +1716,7 @@ class ReformerModelWithLMHead(ReformerPreTrainedModel): Examples:: - from transformers import ReformerModel, ReformerTokenizer + from transformers import ReformerModelWithLMHead, ReformerTokenizer import torch tokenizer = ReformerTokenizer.from_pretrained('google/reformer-crime-and-punishment')