pytorch

72fec96e - fix no shard state dict loading (#120367)

fix no shard state dict loading (#120367) Summary: fix no shard state dict loading Test Plan: CI tests Differential Revision: D51058607 Pull Request resolved: https://github.com/pytorch/pytorch/pull/120367 Approved by: https://github.com/fegin