Apply BaseModelWrapper inside of BaseModelCallback

[DanteOz]

Wrap HuggingFace PreTrainedModel during before_fit() and after_fit() callback hooks. This eliminates the need to explicitly wrap the PreTrainedModel in the low level api.

Tasks:

• [x] Wrap/unwrap model in BaseModelCallback
• [x] Maintain backwards compatibility by checking if model is already wrapped.
• [x] Ensure BaseModelCallback is first callback added when initializing Learner.

[review-notebook-app[bot]]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[ohmeow]

Looking at this today and tomorrow ... sorry for the delay :)

[ohmeow]

Thanks much sir! Yah ... its being merged ... the miracle is happening :)

[ohmeow]

Merged