`kernels-community/flash-attn2` not found, but the build exists

[qgallouedec]

from kernels import get_kernel
get_kernel("kernels-community/flash-attn2")

Traceback (most recent call last):
  File "/fsx/qgallouedec/miniconda3/envs/trl/lib/python3.12/site-packages/kernels/utils.py", line 151, in install_kernel
    return _load_kernel_from_path(repo_path, package_name, variant_locks)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/fsx/qgallouedec/miniconda3/envs/trl/lib/python3.12/site-packages/kernels/utils.py", line 184, in _load_kernel_from_path
    raise FileNotFoundError(
FileNotFoundError: Kernel at path `/fsx/qgallouedec/.cache/huggingface/hub/models--kernels-community--flash-attn2/snapshots/4c6b6ddcdbc5c949a35b4bd968b868bfc802720e` does not have build: torch28-cxx11-cu128-x86_64-linux

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/fsx/qgallouedec/miniconda3/envs/trl/lib/python3.12/site-packages/kernels/utils.py", line 260, in get_kernel
    package_name, package_path = install_kernel(
                                 ^^^^^^^^^^^^^^^
  File "/fsx/qgallouedec/miniconda3/envs/trl/lib/python3.12/site-packages/kernels/utils.py", line 154, in install_kernel
    raise FileNotFoundError(
FileNotFoundError: Kernel `kernels-community/flash-attn2` at revision main does not have build: torch28-cxx11-cu128-x86_64-linux

but

$ pip freeze | grep kernels
kernels==0.11.0

[danieldk]

@drbh flash-attn was renamed to flash-attn2, but the kernel module directories were not renamed, e.g.

https://huggingface.co/kernels-community/flash-attn2/tree/main/build/torch28-cxx11-cu128-x86_64-linux/flash_attn

should be

https://huggingface.co/kernels-community/flash-attn2/tree/main/build/torch28-cxx11-cu128-x86_64-linux/flash_attn2

I'll do a PR to make the error message in kernels nicer, because the current exception is hiding the underlying cause.

[danieldk]

Should work again (hotpatched it on hf.co/kernels-community/flash-attn2).

[qgallouedec]

Thanks!