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 Axolotl Specific Training Args
""" """
from dataclasses import dataclass
from trl import GRPOConfig from trl import GRPOConfig
from axolotl.core.training_args import AxolotlTrainingMixins from axolotl.core.training_args import AxolotlTrainingMixins
class AxolotlGRPOConfig(AxolotlTrainingMixins, GRPOConfig): @dataclass
class AxolotlGRPOConfig(GRPOConfig, AxolotlTrainingMixins):
""" """
Axolotl GRPO Config for GRPO training Axolotl GRPO Config for GRPO training
""" """