RuntimeError: Tensors of the same index must be on the same device and the same dtype except `step` tensors that can be CPU and float32 notwithstanding

I’ve posted this question on StackOverflow already, so I’d like to point you to the StackOverflow webpage for a fix. Thank you very much.