riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 03:01:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Use smaller variant of BLOOM for doc to fix tests

Browse Source
This commit is contained in:
Sylvain Gugger 2022-07-18 15:17:29 -04:00
parent bc8e30bab9
commit 29fd471556
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/bloom/modeling_bloom.py
View File

@ -36,7 +36,7 @@ from .configuration_bloom import BloomConfig
logger = logging.get_logger(__name__) logger = logging.get_logger(__name__)
_CHECKPOINT_FOR_DOC = "bigscience/bloom" _CHECKPOINT_FOR_DOC = "bigscience/bloom-350m"
_CONFIG_FOR_DOC = "BloomConfig" _CONFIG_FOR_DOC = "BloomConfig"
_TOKENIZER_FOR_DOC = "BloomTokenizerFast" _TOKENIZER_FOR_DOC = "BloomTokenizerFast"