Skip to content

Commit 1c8dd9b

Browse files
Fix typo in docstring for teacher_model_init_kwargs (#5260)
1 parent 3fd447a commit 1c8dd9b

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

trl/experimental/gkd/gkd_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ class GKDConfig(SFTConfig):
4242
teacher_model_name_or_path (`str`, *optional*):
4343
Model name or path of the teacher model. If `None`, the teacher model will be the same as the model being
4444
trained.
45-
teacher_model_init_kwargs (`dict[str, Any]]`, *optional*):
45+
teacher_model_init_kwargs (`dict[str, Any]`, *optional*):
4646
Keyword arguments to pass to `AutoModelForCausalLM.from_pretrained` when instantiating the teacher model
4747
from a string.
4848
disable_dropout (`bool`, *optional*, defaults to `True`):

trl/experimental/gold/gold_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ class GOLDConfig(SFTConfig):
4242
teacher_model_name_or_path (`str` or `None`, *optional*, defaults to `None`):
4343
Model name or path of the teacher model. If `None`, the teacher model will be the same as the model being
4444
trained.
45-
teacher_model_init_kwargs (`dict[str, Any]]` or `None`, *optional*, defaults to `None`):
45+
teacher_model_init_kwargs (`dict[str, Any]` or `None`, *optional*, defaults to `None`):
4646
Keyword arguments to pass to `AutoModelForCausalLM.from_pretrained` when instantiating the teacher model
4747
from a string.
4848
teacher_tokenizer_name_or_path (`str` or `None`, *optional*, defaults to `None`):

trl/experimental/minillm/minillm_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ class MiniLLMConfig(GRPOConfig):
2929
arguments, please refer to the [`~transformers.TrainingArguments`] and [`GRPOConfig`] documentation.
3030
3131
Args:
32-
teacher_model_init_kwargs (`dict[str, Any]]`, *optional*):
32+
teacher_model_init_kwargs (`dict[str, Any]`, *optional*):
3333
Keyword arguments to pass to `AutoModelForCausalLM.from_pretrained` when instantiating the teacher model
3434
from a string.
3535
disable_dropout (`bool`, *optional*, defaults to `True`):

0 commit comments

Comments
 (0)