diff --git a/src/axolotl/core/trainers/grpo/args.py b/src/axolotl/core/trainers/grpo/args.py index 3be21a4ce..e5f2cc254 100644 --- a/src/axolotl/core/trainers/grpo/args.py +++ b/src/axolotl/core/trainers/grpo/args.py @@ -1,12 +1,15 @@ """ Axolotl Specific Training Args """ +from dataclasses import dataclass + from trl import GRPOConfig from axolotl.core.training_args import AxolotlTrainingMixins -class AxolotlGRPOConfig(AxolotlTrainingMixins, GRPOConfig): +@dataclass +class AxolotlGRPOConfig(GRPOConfig, AxolotlTrainingMixins): """ Axolotl GRPO Config for GRPO training """