v0.21.2

Bug Fixes

1. Enable torch 2.2.2 (#3161)

Composer currently monkeypatches PyTorch for nightly versions in order to fix upstream bugs. With the release of torch 2.2.2, these monkeypatches were mistakenly applied to the stable release due to incorrect gating on imports. This release fixes the gating, enabling torch 2.2.2.

2. MPS Metric Computation on CPU (#3105)

Due to bugs in computing torchmetrics on Mac devices, we move metric computation onto CPU. This previously had issues with data not properly moving to CPU.

Thank you to @hyenal for this contribution!

3. Batch Sampler Support (#3105)

Composer now supports batch sampler, which previously resulted in an error if specified in the dataloader.

Thank you to @Ghelfi for this contribution!

What's Changed

• Make codequality callable by @mvpatel2000 in #3133
• Explicitly print checkpoint downloading exception by @bigning in #3131
• Change release actions by @mvpatel2000 in #3136
• Passing rank and num_replicas to dist.get_sampler by @ShashankMosaicML in #3137
• Fix broadcast by @mvpatel2000 in #3138
• Compressor fixes by @mbway in #3142
• In case of MPS device also copy batch to CPU by @hyenal in #3105
• Composer object store download retry by @bigning in #3140
• Bump databricks-sdk from 0.22.0 to 0.23.0 by @dependabot in #3144
• Update transformers requirement from !=4.34.0,<4.39,>=4.11 to >=4.11,!=4.34.0,<4.40 by @dependabot in #3148
• Update protobuf requirement from <3.21 to <5.27 by @dependabot in #3147
• Bump traitlets from 5.14.1 to 5.14.2 by @dependabot in #3145
• Bump to 0.21 by @mvpatel2000 in #3150
• Fixing sequence parallel error conditions and adding type float for microbatch_size in typehints by @ShashankMosaicML in #3139
• Fix torch monkeypatch version check by @dakinggg in #3155
• Update torchmetrics requirement from <1.3.2,>=0.10.0 to >=0.10.0,<1.3.3 by @dependabot in #3157
• Bump gitpython from 3.1.42 to 3.1.43 by @dependabot in #3160
• Prevent crash if signal handler cannot be set by @mbway in #3152
• Pin pillow for code quality workflow by @dakinggg in #3162
• Fix torch version check by @dakinggg in #3161
• add more retry to checkpoint downloading by @bigning in #3164
• Append to gpu rank log files instead of throwing error by @jjanezhang in #3166
• Call set_epoch on Dataloader.batch_sampler if defined by @Ghelfi in #3124
• Bump version to 0.21.2 by @mvpatel2000 in #3168

New Contributors

• @hyenal made their first contribution in #3105
• @Ghelfi made their first contribution in #3124

Full Changelog: v0.21.1...v0.21.2