ColossalAI/applications/Chat/coati/experience_buffer/__init__.py

5 lines
133 B
Python

from .base import ExperienceBuffer
from .naive import NaiveExperienceBuffer
__all__ = ['ExperienceBuffer', 'NaiveExperienceBuffer']