Spaces:

flax-community
/

dalle-mini

Running

boris commited on Jul 14, 2021

Commit

bb3bfa6

•

1 Parent(s): a1c047b

feat: use bart large

Files changed (1) hide show

seq2seq/run_seq2seq_flax.py CHANGED Viewed

@@ -84,7 +84,7 @@ MODEL_TYPES = tuple(conf.model_type for conf in MODEL_CONFIG_CLASSES)
 OUTPUT_VOCAB_SIZE = 16384 + 1  # encoded image token space + 1 for bos
 OUTPUT_LENGTH = 256 + 1  # number of encoded tokens + 1 for bos
 BOS_TOKEN_ID = 16384
-BASE_MODEL = 'facebook/bart-large-cnn'
 @dataclass

 OUTPUT_VOCAB_SIZE = 16384 + 1  # encoded image token space + 1 for bos
 OUTPUT_LENGTH = 256 + 1  # number of encoded tokens + 1 for bos
 BOS_TOKEN_ID = 16384
+BASE_MODEL = 'facebook/bart-large'
 @dataclass