ColossalAI/colossalai/zero/legacy/gemini/paramhooks/__init__.py

4 lines
77 B
Python

from ._param_hookmgr import BaseParamHookMgr
__all__ = ["BaseParamHookMgr"]