ColossalAI/applications/Chat/coati/kernels/__init__.py

from .wrapper import convert_to_xformer_model, recover_from_xformer_model

__all__ = [
    "convert_to_xformer_model",
    "recover_from_xformer_model",
]
[chat] add opt attn kernel (#3655) * [chat] add opt attn kernel * [chat] disable xformer during fwd 2023-05-04 08:03:33 +00:00			`from .wrapper import convert_to_xformer_model, recover_from_xformer_model`

			`__all__ = [`
[misc] update pre-commit and run all files (#4752) * [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format 2023-09-19 06:20:26 +00:00			`"convert_to_xformer_model",`
			`"recover_from_xformer_model",`
[chat] add opt attn kernel (#3655) * [chat] add opt attn kernel * [chat] disable xformer during fwd 2023-05-04 08:03:33 +00:00			`]`