Fix the checkpoint for I-BERT (#10994)

bf0840ac · Lysandre Debut · GitHub · ced7284a · bf0840ac
Unverified Commit bf0840ac authored Mar 31, 2021 by Lysandre Debut Committed by GitHub Mar 31, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/ibert/modeling_ibert.py src/transformers/models/ibert/modeling_ibert.py +1 -1

No files found.
--- a/src/transformers/models/ibert/modeling_ibert.py
+++ b/src/transformers/models/ibert/modeling_ibert.py
@@ -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"