Spaces:

flax-community
/

dalle-mini

Running

boris commited on Jul 14, 2021

Commit

ec8d66b

•

1 Parent(s): 1c44a7d

feat: fix typo

Files changed (1) hide show

seq2seq/run_seq2seq_flax.py CHANGED Viewed

@@ -836,7 +836,7 @@ def main():
                 wandb.run.log_artifact(artifact)
             # save to the hub
-            if training_args.push_to_hub
                 model.save_pretrained(
                     training_args.output_dir,
                     params=params,

                 wandb.run.log_artifact(artifact)
             # save to the hub
+            if training_args.push_to_hub:
                 model.save_pretrained(
                     training_args.output_dir,
                     params=params,