make it a dataclass
This commit is contained in:
@@ -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
|
||||||
"""
|
"""
|
||||||
|
|||||||
Reference in New Issue
Block a user