Unverified Commit 68cfffc4 authored by Francisco Kurucz's avatar Francisco Kurucz Committed by GitHub
Browse files

Fix Data2VecTextForCasualLM example code documentation (#20510)

* Fix Data2VecTextForCasualLM example code documentation

* Change RobertaTokenizer to AutoTokenizer in data2vectext example code
parent dd6fb131
...@@ -958,13 +958,13 @@ class Data2VecTextForCausalLM(Data2VecTextPreTrainedModel): ...@@ -958,13 +958,13 @@ class Data2VecTextForCausalLM(Data2VecTextPreTrainedModel):
Example: Example:
```python ```python
>>> from transformers import Data2VecTextTokenizer, Data2VecTextForCausalLM, Data2VecTextConfig >>> from transformers import AutoTokenizer, Data2VecTextForCausalLM, Data2VecTextConfig
>>> import torch >>> import torch
>>> tokenizer = Data2VecTextTokenizer.from_pretrained("facebook/data2vec-text-base") >>> tokenizer = AutoTokenizer.from_pretrained("facebook/data2vec-text-base")
>>> config = Data2VecTextConfig.from_pretrained("data2vec-base") >>> config = Data2VecTextConfig.from_pretrained("facebook/data2vec-text-base")
>>> config.is_decoder = True >>> config.is_decoder = True
>>> model = Data2VecTextForCausalLM.from_pretrained("data2vec-base", config=config) >>> model = Data2VecTextForCausalLM.from_pretrained("facebook/data2vec-text-base", config=config)
>>> inputs = tokenizer("Hello, my dog is cute", return_tensors="pt") >>> inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
>>> outputs = model(**inputs) >>> outputs = model(**inputs)
......
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