From 8e413172508ddd46ceb4fad690f202f36c10b27a Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Thu, 17 Jul 2025 15:31:21 -0400 Subject: [PATCH] don't use include_tokens_per_second for GRPO (#2931) [skip ci] * don't use include_tokens_per_second for GRPO * use blocklist instead --- src/axolotl/core/trainers/grpo/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/core/trainers/grpo/__init__.py b/src/axolotl/core/trainers/grpo/__init__.py index 771f788fe..3e824f705 100644 --- 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: