test

run_mlm_flax_stream.py

@@ -661,11 +661,6 @@ if __name__ == "__main__":
                         params=params
                     )
                     print("Problems pushing this to the hub. The bug should be fixed.")
-            else:
-                model.save_pretrained(
-                    training_args.output_dir,
-                    params=params
-                )
             
         # update tqdm bar
         steps.update(1)