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

6 lines
138 B
Python

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