From 687788954fd0cfbc567fa1202d56a4ff9271944f Mon Sep 17 00:00:00 2001 From: Stella Biderman Date: Thu, 5 Aug 2021 16:15:07 -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 4503f08ff41..6c3787ab1b7 100644 --- a/src/transformers/models/auto/configuration_auto.py +++ b/src/transformers/models/auto/configuration_auto.py @@ -185,6 +185,7 @@ CONFIG_MAPPING = OrderedDict( ("electra", ElectraConfig), ("encoder-decoder", EncoderDecoderConfig), ("flaubert", FlaubertConfig), + ("fsmt", FSMTConfig), ("funnel", FunnelConfig), ("gpt2", GPT2Config), ("gptj", GPTJConfig),