don't use include_tokens_per_second for GRPO (#2931) [skip ci]

* don't use include_tokens_per_second for GRPO * use blocklist instead
2025-07-17 15:31:21 -04:00
parent 9f2bb188a4
commit 8e41317250
1 changed files with 1 additions and 1 deletions
--- a/src/axolotl/core/trainers/grpo/init.py
+++ b/src/axolotl/core/trainers/grpo/init.py
@@ -148,7 +148,7 @@ class GRPOStrategy:

    @classmethod
    def get_blocklist_args_kwargs(cls) -> list[str]:
-        return ["dataset_num_proc", "max_length"]
+        return ["dataset_num_proc", "max_length", "include_tokens_per_second"]

    @classmethod
    def get_reward_func(cls, reward_func_fqn: str) -> RewardFunc: