"git@developer.sourcefind.cn:dadigang/Ventoy.git" did not exist on "f2c94bd963a0296fed93d3170255d4bc13b3bbc1"
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): ...@@ -1515,11 +1515,11 @@ class ReformerModel(ReformerPreTrainedModel):
Examples:: Examples::
from transformers import ReformerModelWithLMHead, ReformerTokenizer from transformers import ReformerModel, ReformerTokenizer
import torch import torch
tokenizer = ReformerTokenizer.from_pretrained('google/reformer-crime-and-punishment') 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 input_ids = torch.tensor(tokenizer.encode("Hello, my dog is cute", add_special_tokens=True)).unsqueeze(0) # Batch size 1
outputs = model(input_ids) outputs = model(input_ids)
...@@ -1716,7 +1716,7 @@ class ReformerModelWithLMHead(ReformerPreTrainedModel): ...@@ -1716,7 +1716,7 @@ class ReformerModelWithLMHead(ReformerPreTrainedModel):
Examples:: Examples::
from transformers import ReformerModel, ReformerTokenizer from transformers import ReformerModelWithLMHead, ReformerTokenizer
import torch import torch
tokenizer = ReformerTokenizer.from_pretrained('google/reformer-crime-and-punishment') 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