Markus Krimmel 6bbc532388 fix: cast the alibi slopes to torch.float32 (#846) пре 9 месеци
..
__init__.py ece539abd6 Add __init__.py files to subdirectories for installation пре 2 година
block.py abbc131173 [LayerNorm] Switch from CUDA to Triton implementation пре 1 година
embedding.py f1a73d0740 Run isort and black on python files пре 1 година
mha.py 6bbc532388 fix: cast the alibi slopes to torch.float32 (#846) пре 9 месеци
mlp.py c3b2196652 Add Alibi to MHA, test with Baichuan-13B пре 1 година