Fixes #87 where model was not being loaded on GPU for inference.
Checklist:
[ ] Was this PR discussed in an issue? It is recommended to first discuss a new feature into a GitHub issue before opening a PR.
[ ] Add tests to cover the fixed bug(s) or the new introduced feature(s) (if appropriate).
[ ] Update the API documentation if a new function is added, or an existing one is deleted. Eventually consider making a new tutorial for new features.
[ ] Write concise and explanatory changelogs below.
[ ] If possible, assign one of the following labels to the PR: feature, fix or test (or ask a maintainer to do it for you).
closes #87
Changelogs
Checklist:
feature,fixortest(or ask a maintainer to do it for you).discussion related to that PR