[DCP][state_dict][doc] Fix the documents for distributed_state_dict (#121276)
As title
Pull Request resolved: https://github.com/pytorch/pytorch/pull/121276
Approved by: https://github.com/wz337, https://github.com/LucasLLC
ghstack dependencies: #121273
