Hello, thank you for your work, but I've encountered some issues. I'm using the ckpt you provided for inference. When I use the slow version of curope, everything works fine, but when I compile curope, the results are problematic, as shown in the image below. I'm using Python 3.10.18, PyTorch 2.4.1, and CUDA 11.8.

Hello, thank you for your work, but I've encountered some issues. I'm using the ckpt you provided for inference. When I use the slow version of curope, everything works fine, but when I compile curope, the results are problematic, as shown in the image below. I'm using Python 3.10.18, PyTorch 2.4.1, and CUDA 11.8.