Sounds similar to - Gpytorch: RuntimeError: expected backend CPU and dtype Double but got backend CPU and dtype Float?