test
Browse files- run_mlm_flax_stream.py +0 -5
run_mlm_flax_stream.py
CHANGED
|
@@ -661,11 +661,6 @@ if __name__ == "__main__":
|
|
| 661 |
params=params
|
| 662 |
)
|
| 663 |
print("Problems pushing this to the hub. The bug should be fixed.")
|
| 664 |
-
else:
|
| 665 |
-
model.save_pretrained(
|
| 666 |
-
training_args.output_dir,
|
| 667 |
-
params=params
|
| 668 |
-
)
|
| 669 |
|
| 670 |
# update tqdm bar
|
| 671 |
steps.update(1)
|
|
|
|
| 661 |
params=params
|
| 662 |
)
|
| 663 |
print("Problems pushing this to the hub. The bug should be fixed.")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 664 |
|
| 665 |
# update tqdm bar
|
| 666 |
steps.update(1)
|