mirror of https://github.com/hpcaitech/ColossalAI
3c91e3f176
* adapt to baichuan2 13B * adapt to baichuan2 13B * change BAICHUAN_MODEL_NAME_OR_PATH * fix test_decoding_attn.py * Modifications based on review comments. * change BAICHUAN_MODEL_NAME_OR_PATH * mv attn mask processes to test flash decoding * mv get_alibi_slopes baichuan modeling * fix bugs in test_baichuan.py |
||
---|---|---|
.. | ||
jit | ||
triton | ||
__init__.py | ||
extensions | ||
kernel_loader.py |