CUDA not working on 0.2.1

[nammerduong]

Ok so when I first used

pip3 install mokuro

I think it correctly installed the latest version of mokuro (0.2.1) but I wasn't able to use CUDA. So I started uninstalling old CUDA toolkit and new pythons. But then mokuro stopped working so I had to use

pip3 install mokuro

again. But this time I was stuck with the version of mokuro that doesn't create a .mokuro file.

Then I used

pip3 install git+https://github.com/kha-white/mokuro.git@web-reader

to get a mokuro that creates .mokuro file and can use CUDA. Then I tried

pip3 install git+https://github.com/kha-white/mokuro.git

and was able to get the latest 0.2.1 version of mokuro but it still can't use CUDA for some reason so I have to stick with the web-reader branch to be able to use CUDA.

I'm using:
Python: 3.10.11
Git: 2.47.1
CUDA Toolkit: 11.8
Pytorch: 2.5.1