Fix stop tokens in PPO (#2304)

pytorch · Jan 31, 2025 · 6487029 · 6487029
1 parent afd5bc4
commit 6487029
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recipes/ppo_full_finetune_single_device.py b/recipes/ppo_full_finetune_single_device.py
@@ -357,7 +357,7 @@ def _setup_training_hyperparameters(self, cfg) -> None:
                     "This may lead to unexpected behaviour."
                 )
         else:
-            if not hasattr(self._tokenizer.stop_tokens):
+            if not hasattr(self._tokenizer, "stop_tokens"):
                 warn(
                     "No stop tokens defined in tokenizer, and no stop_token_ids provided. This may lead to unexpected behaviour."
                 )