Thanks!
Based on the code it seems that you might be creating numpy arrays for your input data, which uses float64 by default as the dtype, before transforming them to tensors.
Could you transform the input tensors for float32 via tensor = tensor.float() before passing them to the model and rerun the code?