Update tokenizer_utils.py

This commit is contained in:
Daniel Han 2025-02-09 19:19:51 -08:00
commit 16ddec731a

View file

@ -1059,8 +1059,11 @@ for trainer_name in ("SFTTrainer", "DPOTrainer", "KTOTrainer"):
if trainer_text is None: continue
try:
exec(trainer_text, globals())
except:
raise RuntimeError(f"Unsloth: Please file a bug report! Error patching {trainer_name}")
except Exception as error:
raise RuntimeError(
f"Unsloth: Please file a bug report! Error patching {trainer_name}. Error:\n"\
f"{str(error)}",
)
exec(f"trl.trainer.{trainer_name} = Unsloth{trainer_name}", globals())
pass