diff --git a/src/transformers/kernels/__init__.py b/src/transformers/kernels/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/utils/check_build.py b/utils/check_build.py index e3cca31f8..9ac309bd6 100644 --- a/utils/check_build.py +++ b/utils/check_build.py @@ -23,6 +23,9 @@ FILES_TO_FIND = [ "kernels/rwkv/wkv_op.cpp", "kernels/deformable_detr/ms_deform_attn.h", "kernels/deformable_detr/cuda/ms_deform_im2col_cuda.cuh", + "kernels/falcon_mamba/selective_scan_with_ln_interface.py", + "kernels/falcon_mamba/__init__.py", + "kernels/__init__.py", "models/graphormer/algos_graphormer.pyx", ] diff --git a/utils/check_inits.py b/utils/check_inits.py index a9a841f3b..840bad086 100644 --- a/utils/check_inits.py +++ b/utils/check_inits.py @@ -333,6 +333,7 @@ IGNORE_SUBMODULES = [ "safetensors_conversion", "modeling_gguf_pytorch_utils", "kernels.falcon_mamba", + "kernels", ]