adding 'reward_processing_classes'

This commit is contained in:
Salman Mohammadi
2025-02-05 18:13:05 +00:00
parent b8f258817e
commit 65b6519447

View File

@@ -53,6 +53,10 @@ class GRPOStrategy:
for reward_func_fqn in cfg.grpo_reward_funcs:
reward_funcs.append(cls.get_reward_func(reward_func_fqn))
trainer_kwargs["reward_funcs"] = reward_funcs
if cfg.grpo_reward_processing_classes:
trainer_kwargs[
"reward_processing_classes"
] = cfg.grpo_reward_processing_classes
return trainer_kwargs
@classmethod