From 057aad6b4d6c033708ba33178b230cd8d1e2f3f2 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 10 Mar 2025 04:58:59 -0700 Subject: [PATCH] Update rl.py --- unsloth/models/rl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index 5cb76ae1d1..da4225e88a 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -349,7 +349,7 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"): "elif isinstance(data_collator, DataCollatorForLanguageModeling) and 'labels' in train_dataset.column_names):\n"\ " print('Unsloth: Changing data collator to `DataCollatorForSeq2Seq` since `labels` found.)\n"\ " data_collator = DataCollatorForSeq2Seq("\ - "tokenizer = processing_class if 'processing_class' in locals() else tokenizer)\n"\ + "tokenizer = processing_class if 'processing_class' in locals() else tokenizer)\n" extra_args += data_collator_check pass