RuntimeWarning: invalid value encountered in cast and blank inference

[rolazazz]

I'm getting the warning below if I run the sample code on my laptop with GTX 1650 Ti

C:\Repos\BEN2\.venv\src\ben2\src\ben2\modeling_ben2.py:1557: RuntimeWarning: invalid value encountered in cast
im_array = (result * 255).permute(1, 2, 0).cpu().data.numpy().astype(np.uint8)

No errors with the same code on a workstation with RTX 2060.

I've also tried loading the model on CPU and i'm getting an exception on both systems:

RuntimeError: Input type (struct c10::Half) and bias type (float) should be the same