From 64e07fc306ad3ea22126df7bc01598d8a1fae39c Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 3 Jun 2025 00:07:52 -0700 Subject: [PATCH] Update rl.py --- unsloth/models/rl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index e02648a2dd..3fa3fe7132 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -349,9 +349,9 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"): " max_length = args.max_length\n"\ " else:\n"\ " model_max_length = getattr(model, 'max_seq_length', None)\n"\ - " print(model_max_length, 'mml1')\n"\ + " # print(model_max_length, 'mml1')\n"\ " if model_max_length is None: model_max_length = getattr(model, 'max_length', None)\n"\ - " print(model_max_length, 'mml2')\n"\ + " # print(model_max_length, 'mml2')\n"\ " if model_max_length is not None:\n"\ " args.max_length = model_max_length\n"\ " max_length = args.max_length\n"\