make it a dataclass

This commit is contained in:
Wing Lian
2025-02-03 00:45:58 -05:00
parent b1c4711145
commit 300ffc2cb6

View File

@@ -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
"""