Unverified Commit 99d1a694 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

fix examples (#4192)

parent 74ffc9ea
......@@ -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')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment