InternLM/internlm/solver/optimizer/__init__.py

7 lines
183 B
Python

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
from .hybrid_zero_optim import HybridZeroOptimizer, reload_zero_fp32_buff
__all__ = ["HybridZeroOptimizer", "reload_zero_fp32_buff"]