ColossalAI/applications/ChatGPT/chatgpt/dataset/__init__.py

5 lines
113 B
Python

from .reward_dataset import RewardDataset
from .utils import is_rank_0
__all__ = ['RewardDataset', 'is_rank_0']