Skip to content

[v0.23.2] Patch: Support `max_shard_size` as string in `split_state_dict_into_shards_factory`
Compare
Choose a tag to compare
@Wauplin Wauplin released this 27 May 08:32
· 196 commits to main since this release
v0.23.2
f3b9c67
This commit was signed with the committer’s verified signature.
Wauplin Lucain
GPG key ID: 9838FE02BECE1A02
Learn about vigilant mode.

split_state_dict_into_shards_factory now accepts string values as max_shard_size (ex: "5MB"), in addition to integer values. Related PR: #2286.

Full Changelog: v0.23.1...v0.23.2

Assets 2
Loading