Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "ce111feed139d3e2a05bb364ad8ee186093ad3c7"
Unverified Commit fc3551a6 authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

add flax mbart in auto seq2seq lm (#13560)

parent 3081d386
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/models/auto/modeling_flax_auto.py
View file @ fc3551a6
...@@ -81,6 +81,7 @@ FLAX_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING_NAMES = OrderedDict( ...@@ -81,6 +81,7 @@ FLAX_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING_NAMES = OrderedDict(
[ [
# Model for Seq2Seq Causal LM mapping # Model for Seq2Seq Causal LM mapping
("bart", "FlaxBartForConditionalGeneration"), ("bart", "FlaxBartForConditionalGeneration"),
("mbart", "FlaxMBartForConditionalGeneration"),
("t5", "FlaxT5ForConditionalGeneration"), ("t5", "FlaxT5ForConditionalGeneration"),
("mt5", "FlaxMT5ForConditionalGeneration"), ("mt5", "FlaxMT5ForConditionalGeneration"),
("marian", "FlaxMarianMTModel"), ("marian", "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