Pytorch NLP model doesn’t use GPU when making inference

[redrussianarmy]

Thank you valhalla. It works on GPU now.