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

4 lines
77 B
Python

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