Garrett Goon 390f153469 Add padding-free to bamba (#35861) ... * add seq_idx and fa kwargs * update tests * docs and grad ckpt support * fmt * better names * test_raise_missing_padding_free_kwarg_errs * + seq_idx in doc strings * padding free training docs * add link to pr plots * raise err on attn_mask with padding free * rm raising missing padding free err test * BambaFlashAttentionKwargs * run modular util for modular_granitemoehybrid.py		2025-05-20 17:13:59 +02:00
..
__init__.py	Add the Bamba Model (#34982)	2024-12-18 20:18:17 +01:00
test_modeling_bamba.py	Add padding-free to bamba (#35861)	2025-05-20 17:13:59 +02:00