GitLab

* style: Added typing in models._utils

* fix: Removed non-necessary import

* fix: Removed type annotation of forward method

* refactor: Removed un-necessary import