Fix 'initialized after field' error (#30908)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/30908
Same as title.
Test Plan: Wait for CI to clear.
Reviewed By: bddppq, xw285cornell
Differential Revision: D18862837
fbshipit-source-id: bc34356b85774fc20ba46d321c8a2bb5d5c727f6