From cfaaae4c4dc70f1fbe9abd60fc8bd0b863b8c011 Mon Sep 17 00:00:00 2001 From: Stella Biderman Date: Thu, 5 Aug 2021 16:17:31 -0400 Subject: [PATCH] Update configuration_auto.py --- src/transformers/models/auto/configuration_auto.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/transformers/models/auto/configuration_auto.py b/src/transformers/models/auto/configuration_auto.py index 6c3787ab1b7..5c925d6feaf 100644 --- a/src/transformers/models/auto/configuration_auto.py +++ b/src/transformers/models/auto/configuration_auto.py @@ -191,6 +191,7 @@ CONFIG_MAPPING = OrderedDict( ("gptj", GPTJConfig), ("gpt_neo", GPTNeoConfig), ("hubert", HubertConfig), + ("ibert", IBertConfig), ("layoutlm", LayoutLMConfig), ("led", LEDConfig), ("longformer", LongformerConfig),