From 23ba1ca3ac06ac8c05e54ed0c69cdb4a9c7645b4 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 11 Feb 2025 23:58:26 -0800 Subject: [PATCH] Update rl_replacements.py --- unsloth/models/rl_replacements.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/rl_replacements.py b/unsloth/models/rl_replacements.py index 65138feb13..ba759095e8 100644 --- a/unsloth/models/rl_replacements.py +++ b/unsloth/models/rl_replacements.py @@ -33,6 +33,7 @@ def sft_trainer_fix_untraiend_tokens(call_args, extra_args): "if 'tokenizer' not in locals(): tokenizer = processing_class\n"\ "fix_untrained_tokens(model, tokenizer, train_dataset, IGNORED_TOKENIZER_NAMES, eps = 1e-16)\n"\ "fix_zero_training_loss(model, tokenizer, train_dataset)\n" + "print(1111)\n", return fix_tokenizer return "" pass