diff --git a/src/transformers/models/bloom/modeling_bloom.py b/src/transformers/models/bloom/modeling_bloom.py index 9f520ae473bf..d49d6117d338 100644 --- a/src/transformers/models/bloom/modeling_bloom.py +++ b/src/transformers/models/bloom/modeling_bloom.py @@ -38,7 +38,7 @@ _CHECKPOINT_FOR_DOC = "bigscience/Bloom" _CONFIG_FOR_DOC = "BloomConfig" -_TOKENIZER_FOR_DOC = "BloomTokenizer" +_TOKENIZER_FOR_DOC = "BloomTokenizerFast" BLOOM_PRETRAINED_MODEL_ARCHIVE_LIST = [ "bigscience/bigscience-small-testing", @@ -626,7 +626,7 @@ def _set_gradient_checkpointing(self, module, value=False): If `past_key_values` is used, only `input_ids` that do not have their past calculated should be passed as `input_ids`. - Indices can be obtained using [`BloomTokenizer`]. See [`PreTrainedTokenizer.encode`] and + Indices can be obtained using [`BloomTokenizerFast`]. See [`PreTrainedTokenizer.encode`] and [`PreTrainedTokenizer.__call__`] for details. [What are input IDs?](../glossary#input-ids)