accelerate
418015aa - test: add regression test for no_split_module_classes accepting set type (#4048)

Commit
93 days ago
test: add regression test for no_split_module_classes accepting set type (#4048) * test: add regression test for no_split_module_classes accepting set type get_balanced_memory and infer_auto_device_map should accept a set for no_split_module_classes without raising 'TypeError: unhashable type: set'. Bug was present in <=1.5.0, fixed in #2345, this test prevents regression. * Apply style fixes --------- Co-authored-by: 你的GitHub名稱 <你的GitHub信箱> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Author
Parents
Loading