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"]