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
