ColossalAI/applications/ChatGPT/chatgpt/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']