mirror of https://github.com/hpcaitech/ColossalAI
![]() * support npu * support pretrain support pretrain fix * support lora fix fix * support chatglm fix fxi fix [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci fix fix [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci fix [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci fix fix fix * Update train.py * Update train.py |
||
---|---|---|
.. | ||
__init__.py | ||
flash_attention_dao_cuda.py | ||
flash_attention_npu.py | ||
flash_attention_sdpa_cuda.py |