pytorch

6c858e37 - [FSDP][Easy] Remove unneeded `TrainingState` transition (#88232)

[FSDP][Easy] Remove unneeded `TrainingState` transition (#88232) Follow-up from previous PR in the stack Pull Request resolved: https://github.com/pytorch/pytorch/pull/88232 Approved by: https://github.com/mrshenli