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

7 lines
173 B
Python

from .gpt_actor import GPTActor
from .gpt_critic import GPTCritic
from .gpt_lm import GPTLM
from .gpt_rm import GPTRM
__all__ = ['GPTActor', 'GPTCritic', 'GPTRM', 'GPTLM']