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

6 lines
136 B
Python

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