What's Changed
- Removed extraneous usage of
fsdp_config.load_monolith_rank0_onlysince that's unreliable by @rithwik-db in #3901 - Fixed automicrobatching issue for FSDP1 by @rithwik-db in #3909
- reverted mlflow upgrade due to slowdowns by @ethantang-db in #3910
Full Changelog: v0.32.0...v0.32.1
Contributors
rithwik-db and ethantang-db