GitLab

* Don't copy when unwrapping model.

Otherwise an exception is raised when using fp16.

* Remove unused import