accelerate
b7d5d907 - adjust overriding of model's forward function (#1492)

Commit
2 years ago
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>
Author
Parents
Loading