SueJane 3f1b4d38e7 Fix: check the type of max_seqlen_k instead of checking max_seqlen twice (#1127) vor 4 Monaten
..
__init__.py ece539abd6 Add __init__.py files to subdirectories for installation vor 2 Jahren
block.py abbc131173 [LayerNorm] Switch from CUDA to Triton implementation vor 11 Monaten
embedding.py f1a73d0740 Run isort and black on python files vor 1 Jahr
mha.py 3f1b4d38e7 Fix: check the type of max_seqlen_k instead of checking max_seqlen twice (#1127) vor 4 Monaten
mlp.py c3b2196652 Add Alibi to MHA, test with Baichuan-13B vor 11 Monaten