Hi. It would be better if you provide error stack, to check from what point the problem is coming. Also, the general advice is to explicitly check if both your model and your input are on the same device.