Xunzhuo's picture
Add System One typed batch API and matching fine-tuning inputs
a2af77f verified
Raw History Blame
282 Bytes
from .profile import Complete1KCollator, MAX_INPUT_TOKENS, predict_1k
from ._auto import predict_auto_1k
from ._system_one import SystemOne, system_one_records
__all__ = ["Complete1KCollator", "MAX_INPUT_TOKENS", "predict_1k", "predict_auto_1k", "SystemOne", "system_one_records"]