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

Fix the checkpoint for I-BERT (#10994)

Browse Source
This commit is contained in:
Lysandre Debut 2021-03-31 08:02:51 -04:00 committed by GitHub
parent ced7284a60
commit bf0840accc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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/ibert/modeling_ibert.py
View File

@ -43,7 +43,7 @@ from .quant_modules import IntGELU, IntLayerNorm, IntSoftmax, QuantAct, QuantEmb
logger = logging.get_logger(__name__)
_CHECKPOINT_FOR_DOC = "ibert-roberta-base"
_CHECKPOINT_FOR_DOC = "kssteven/ibert-roberta-base"
_CONFIG_FOR_DOC = "IBertConfig"
_TOKENIZER_FOR_DOC = "RobertaTokenizer"