pytorch

0811f152 - [DCP][state_dict] Let _offload_state_dict_to_cpu to return the companion_obj if it exist. (#121273)

[DCP][state_dict] Let _offload_state_dict_to_cpu to return the companion_obj if it exist. (#121273) As title Pull Request resolved: https://github.com/pytorch/pytorch/pull/121273 Approved by: https://github.com/wz337, https://github.com/LucasLLC