From ea2da65e15591b423a3a9491c13512278d609926 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 26 Jun 2025 00:27:17 -0700 Subject: [PATCH] Update rl_replacements.py --- unsloth/models/rl_replacements.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/unsloth/models/rl_replacements.py b/unsloth/models/rl_replacements.py index 0056f27509..e44005cedc 100644 --- a/unsloth/models/rl_replacements.py +++ b/unsloth/models/rl_replacements.py @@ -251,8 +251,7 @@ def grpo_trainer__get_per_token_logps(function_name, function): if function_name != "_get_per_token_logps": return function def _get_per_token_logps(self, model, input_ids, attention_mask, logits_to_keep): - if True: #os.environ.get('UNSLOTH_USE_NEW_MODEL', '0') == '0': - print("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!") + if os.environ.get('UNSLOTH_USE_NEW_MODEL', '0') == '0': return None # Unsloth efficient GRPO # Otherwise, calculate normally: if not hasattr(self, '_autocast_dtype'):