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

7 lines
161 B
Python

from .actor import Actor
from .critic import Critic
from .lm import LM
from .reward_model import RewardModel
__all__ = ['Actor', 'Critic', 'RewardModel', 'LM']