Fix checkpoints for Gaudi

[Jianhong-Zhang]

remove _orig_mod prefix from checkpoints from torch.compile trained
model

Signed-off-by: Jianhong-Zhang jianhong.zhang@intel.com

[mergify]

This pull request has merge conflicts that must be resolved before it can be
merged. @Jianhong-Zhang please rebase it. https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/working-with-forks/syncing-a-fork

[github-actions]

This pull request has been automatically marked as stale because it has not had activity within 90 days. It will be automatically closed if no further activity occurs within 30 days.

[mergify]

This pull request has merge conflicts that must be resolved before it can be
merged. @Jianhong-Zhang please rebase it. https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/working-with-forks/syncing-a-fork

[github-actions]

This pull request has been automatically marked as stale because it has not had activity within 90 days. It will be automatically closed if no further activity occurs within 30 days.

[github-actions]

This pull request has been automatically closed due to inactivity. Please feel free to reopen if you intend to continue working on it!