SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix: Fix FalconMamba training issues due to incompatible kernels (#33195)

Browse Source 
* fix FM training kernels

* fix copies

* fix copies

* propagate to slow path

* make it BC

* add comment

* fix test
This commit is contained in:
Younes Belkada
2024-09-05 13:55:08 +04:00
committed by GitHub
parent 43df47d8e7
commit 47b096412d
6 changed files with 597 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
utils/check_inits.py
View File

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