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',
]