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

7 lines
155 B
Python

from .wrapper import convert_to_xformer_model, recover_from_xformer_model
__all__ = [
"convert_to_xformer_model",
"recover_from_xformer_model",
]