ColossalAI/applications/Chat/coati/models/opt/__init__.py

7 lines
173 B
Python

from .opt_actor import OPTActor
from .opt_critic import OPTCritic
from .opt_lm import OPTLM
from .opt_rm import OPTRM
__all__ = ['OPTActor', 'OPTCritic', 'OPTRM', 'OPTLM']