.. |
__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 1 Jahr |
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 6 Monaten |
mlp.py
|
c3b2196652
Add Alibi to MHA, test with Baichuan-13B
|
vor 1 Jahr |