pytorch

056cfb04 - Revert "[ONNX] Update training state logic to support ScriptedModule (#86745)"

Revert "[ONNX] Update training state logic to support ScriptedModule (#86745)" This reverts commit 960b98128e475b15b66119f325232039799852cd. Reverted https://github.com/pytorch/pytorch/pull/86745 on behalf of https://github.com/janeyx99 due to https://hud.pytorch.org/pytorch/pytorch/commit/960b98128e475b15b66119f325232039799852cd broke onnx tests on trunk