pytorch

0f3f4f55 - Revert "[nit][DCP][DSD] Remove Unused Variables in test_state_dict.py (#121204)"

Revert "[nit][DCP][DSD] Remove Unused Variables in test_state_dict.py (#121204)" This reverts commit 4186c365313e909dfc8574c4469e5015439c2924. Reverted https://github.com/pytorch/pytorch/pull/121204 on behalf of https://github.com/huydhn due to Sorry for reverting your PR, but the failure looks legit ([comment](https://github.com/pytorch/pytorch/pull/121204#issuecomment-1986252526))