saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix import of FalconMambaForCausalLM (#33381)

Browse source 
* fix build issues with FM kernels

* try another approach

* test

* fix

* add init files

* push fix

* fix

* fixup

* fix duplicate

* fix

* fix

* fix
This commit is contained in:
Younes Belkada 2024-09-10 11:14:54 +04:00 committed by GitHub
parent f745e7d3f9
commit 7f112caac2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

0
src/transformers/kernels/__init__.py Normal file
View file

3
utils/check_build.py
View file

@ -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",
]

1
utils/check_inits.py
View file

@ -333,6 +333,7 @@ IGNORE_SUBMODULES = [
"safetensors_conversion",
"modeling_gguf_pytorch_utils",
"kernels.falcon_mamba",
"kernels",
]