fix: GRPO config not accept max_prompt_length (#3390) [skip ci]

2026-02-10 17:52:09 +07:00
parent b6d3653f74
commit ed7105dba7
1 changed files with 6 additions and 1 deletions
--- a/src/axolotl/core/trainers/grpo/init.py
+++ b/src/axolotl/core/trainers/grpo/init.py
@@ -164,7 +164,12 @@ class GRPOStrategy:

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

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