transformers
ff49f7c4 - Fix NaN weights on non-rank-0 FSDP processes (#45050)

Commit

15 days ago

Fix NaN weights on non-rank-0 FSDP processes (#45050) Init with zeros instead of empty in _move_missing_keys_from_meta_to_device

References

Author

albertvillanova

Parents