Skip to content

Commit 3fd447a

Browse files
Support JSON string parsing of teacher_model_init_kwargs in MiniLLMConfig (#5259)
1 parent 41602bc commit 3fd447a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

trl/experimental/minillm/minillm_config.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,8 @@ class MiniLLMConfig(GRPOConfig):
4747
Whether to apply length normalization to the rewards.
4848
"""
4949

50+
_VALID_DICT_FIELDS = TrainingArguments._VALID_DICT_FIELDS + ["teacher_model_init_kwargs"]
51+
5052
teacher_model_init_kwargs: dict[str, Any] | None = field(
5153
default=None,
5254
metadata={

0 commit comments

Comments
 (0)