[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-03-10 16:04:44 +00:00 committed by Daniel Han
commit 39179cda35

View file

@ -155,6 +155,7 @@ def _patch_resume_from_checkpoint_memory(trainer_class):
_unsloth_train_with_resume_guard._unsloth_resume_guard = True
trainer_class.train = _unsloth_train_with_resume_guard
def PatchRL(FastLanguageModel):
try:
from trl.models.utils import unwrap_model_for_generation