Spaces:

flax-community
/

dalle-mini

Running

boris commited on Nov 28, 2021

Commit

2be9847

•

1 Parent(s): 80b41d1

feat: don't ignore mismatched

Files changed (1) hide show

dev/seq2seq/run_seq2seq_flax.py CHANGED Viewed

@@ -531,7 +531,6 @@ def main():
                 config=config,
                 seed=training_args.seed_model,
                 dtype=getattr(jnp, model_args.dtype),
-                ignore_mismatched_sizes=True,
             )
             # avoid OOM on TPU: see https://github.com/google/flax/issues/1658
             print(model.params)

                 config=config,
                 seed=training_args.seed_model,
                 dtype=getattr(jnp, model_args.dtype),
             )
             # avoid OOM on TPU: see https://github.com/google/flax/issues/1658
             print(model.params)