search

Adamdad / kat

Kolmogorov-Arnold Transformer: A PyTorch Implementation with CUDA kernel
MIT License
242 stars 12 forks source link

in katransformer.py, the line from kat_rational import KAT_Group fails to find the corresponding file #4

Open Cyc-InGitHub opened 1 hour ago

Cyc-InGitHub commented 1 hour ago

Thank you very much for your sharing. However, I encountered an issue while using it: in katransformer.py, the line from kat_rational import KAT_Group fails to find the corresponding file. I'm wondering if you've experienced the same issue.

Adamdad commented 1 hour ago

Hello @Cyc-InGitHub ,Have you install the CUDA extension already? kat_rational can be imported once the CUDA extension is installed https://github.com/Adamdad/rational_kat_cu