Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "2f32dfd33b8a0a557d84caf479663b21afab4a2b"
Unverified Commit 51eb6d34 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

[Flax] Fix mt5 auto (#12612) 

* fix_torch_device_generate_test

* remove @

* fix mt5 auto
parent e7f33e8c
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
src/transformers/models/auto/modeling_flax_auto.py
View file @ 51eb6d34
...@@ -84,6 +84,7 @@ from .configuration_auto import ( ...@@ -84,6 +84,7 @@ from .configuration_auto import (
GPTNeoConfig, GPTNeoConfig,
MarianConfig, MarianConfig,
MBartConfig, MBartConfig,
MT5Config,
RobertaConfig, RobertaConfig,
T5Config, T5Config,
ViTConfig, ViTConfig,
...@@ -108,6 +109,7 @@ FLAX_MODEL_MAPPING = OrderedDict( ...@@ -108,6 +109,7 @@ FLAX_MODEL_MAPPING = OrderedDict(
(ViTConfig, FlaxViTModel), (ViTConfig, FlaxViTModel),
(MBartConfig, FlaxMBartModel), (MBartConfig, FlaxMBartModel),
(T5Config, FlaxT5Model), (T5Config, FlaxT5Model),
(MT5Config, FlaxT5Model),
(Wav2Vec2Config, FlaxWav2Vec2Model), (Wav2Vec2Config, FlaxWav2Vec2Model),
(MarianConfig, FlaxMarianModel), (MarianConfig, FlaxMarianModel),
] ]
...@@ -123,6 +125,7 @@ FLAX_MODEL_FOR_PRETRAINING_MAPPING = OrderedDict( ...@@ -123,6 +125,7 @@ FLAX_MODEL_FOR_PRETRAINING_MAPPING = OrderedDict(
(ElectraConfig, FlaxElectraForPreTraining), (ElectraConfig, FlaxElectraForPreTraining),
(MBartConfig, FlaxMBartForConditionalGeneration), (MBartConfig, FlaxMBartForConditionalGeneration),
(T5Config, FlaxT5ForConditionalGeneration), (T5Config, FlaxT5ForConditionalGeneration),
(MT5Config, FlaxT5ForConditionalGeneration),
(Wav2Vec2Config, FlaxWav2Vec2ForPreTraining), (Wav2Vec2Config, FlaxWav2Vec2ForPreTraining),
] ]
) )
...@@ -144,6 +147,7 @@ FLAX_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING = OrderedDict( ...@@ -144,6 +147,7 @@ FLAX_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING = OrderedDict(
# Model for Seq2Seq Causal LM mapping # Model for Seq2Seq Causal LM mapping
(BartConfig, FlaxBartForConditionalGeneration), (BartConfig, FlaxBartForConditionalGeneration),
(T5Config, FlaxT5ForConditionalGeneration), (T5Config, FlaxT5ForConditionalGeneration),
(MT5Config, FlaxT5ForConditionalGeneration),
(MarianConfig, FlaxMarianMTModel), (MarianConfig, FlaxMarianMTModel),
] ]
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment