mirror of https://github.com/hpcaitech/ColossalAI
7bd0bee8ea
* [chat] add opt attn kernel * [chat] disable xformer during fwd |
||
---|---|---|
.. | ||
__init__.py | ||
opt_attn.py | ||
wrapper.py |