Spaces:

tom-doerr
/

logo_generator

Runtime error

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)