Update __init__.py

This commit is contained in:
Daniel Han 2024-11-13 17:38:26 -08:00
commit a97ba4c6a4

View file

@ -177,5 +177,5 @@ from .chat_templates import *
from .tokenizer_utils import *
from .trainer import *
# patch sft trainer
_patch_trl_trainer()
# Patch TRL trainers for backwards compatibility
_patch_trl_trainer()