adjust overriding of model's forward function (#1492)
* adjust overriding of model's forward function
* bug fix
* extend solution to all model.forward overrides
* leave fp8 section alone
* make style
---------
Co-authored-by: root <root@orttrainingdev8.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>
Co-authored-by: Prathik Rao <prathikrao@microsoft.com@orttrainingdev8.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>