diff --git a/trl/trainer/orpo_trainer.py b/trl/trainer/orpo_trainer.py index f84d4fcbb7..dfc78fbee2 100644 --- a/trl/trainer/orpo_trainer.py +++ b/trl/trainer/orpo_trainer.py @@ -1129,10 +1129,10 @@ def create_model_card( Creates a draft of a model card using the information available to the `Trainer`. Args: - model_name (`str`, *optional*, defaults to `None`): - The name of the model. - dataset_name (`str`, *optional*, defaults to `None`): - The name of the dataset used for training. + model_name (`str` or `None`, *optional*, defaults to `None`): + Name of the model. + dataset_name (`str` or `None`, *optional*, defaults to `None`): + Name of the dataset used for training. tags (`str`, `list[str]` or `None`, *optional*, defaults to `None`): Tags to be associated with the model card. """