Add FlaxEncoderDecoderModel to modeling_flax_auto.py

huggingface · Aug 13, 2021 · b6f5f4f · b6f5f4f
1 parent 281fa2a
commit b6f5f4f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/transformers/models/auto/modeling_flax_auto.py b/src/transformers/models/auto/modeling_flax_auto.py
@@ -79,6 +79,7 @@
         ("t5", "FlaxT5ForConditionalGeneration"),
         ("mt5", "FlaxMT5ForConditionalGeneration"),
         ("marian", "FlaxMarianMTModel"),
+        ("encoder-decoder", "FlaxEncoderDecoderModel"),
     ]
 )