From ab5dac06ecc8a5ccbaf6905479c0f1d8a1899c11 Mon Sep 17 00:00:00 2001 From: Juan Francisco Kurucz Date: Wed, 30 Nov 2022 13:12:45 -0300 Subject: [PATCH 1/2] Fix Data2VecTextForCasualLM example code documentation --- .../models/data2vec/modeling_data2vec_text.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/transformers/models/data2vec/modeling_data2vec_text.py b/src/transformers/models/data2vec/modeling_data2vec_text.py index 3ba0ab1c48311..7b5436bb9dd23 100644 --- a/src/transformers/models/data2vec/modeling_data2vec_text.py +++ b/src/transformers/models/data2vec/modeling_data2vec_text.py @@ -958,13 +958,13 @@ def forward( Example: ```python - >>> from transformers import Data2VecTextTokenizer, Data2VecTextForCausalLM, Data2VecTextConfig + >>> from transformers import RobertaTokenizer, Data2VecTextForCausalLM, Data2VecTextConfig >>> import torch - >>> tokenizer = Data2VecTextTokenizer.from_pretrained("facebook/data2vec-text-base") - >>> config = Data2VecTextConfig.from_pretrained("data2vec-base") + >>> tokenizer = RobertaTokenizer.from_pretrained("facebook/data2vec-text-base") + >>> config = Data2VecTextConfig.from_pretrained("facebook/data2vec-text-base") >>> 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") >>> outputs = model(**inputs) From 62e16afb608e509e0f2d55b82a341cb9ba9b55f7 Mon Sep 17 00:00:00 2001 From: Juan Francisco Kurucz Date: Wed, 30 Nov 2022 16:16:38 -0300 Subject: [PATCH 2/2] Change RobertaTokenizer to AutoTokenizer in data2vectext example code --- src/transformers/models/data2vec/modeling_data2vec_text.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/data2vec/modeling_data2vec_text.py b/src/transformers/models/data2vec/modeling_data2vec_text.py index 7b5436bb9dd23..0750f1ca5892d 100644 --- a/src/transformers/models/data2vec/modeling_data2vec_text.py +++ b/src/transformers/models/data2vec/modeling_data2vec_text.py @@ -958,10 +958,10 @@ def forward( Example: ```python - >>> from transformers import RobertaTokenizer, Data2VecTextForCausalLM, Data2VecTextConfig + >>> from transformers import AutoTokenizer, Data2VecTextForCausalLM, Data2VecTextConfig >>> import torch - >>> tokenizer = RobertaTokenizer.from_pretrained("facebook/data2vec-text-base") + >>> tokenizer = AutoTokenizer.from_pretrained("facebook/data2vec-text-base") >>> config = Data2VecTextConfig.from_pretrained("facebook/data2vec-text-base") >>> config.is_decoder = True >>> model = Data2VecTextForCausalLM.from_pretrained("facebook/data2vec-text-base", config=config)