Skip to content

Commit

Permalink
chore: lint
Browse files Browse the repository at this point in the history
  • Loading branch information
winglian committed Feb 4, 2025
1 parent cf30719 commit 9684437
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions trl/trainer/grpo_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ class GRPOConfig(TrainingArguments):
improve the model's throughput. However, if the value is too high, it may cause out-of-memory (OOM) errors
during initialization.
vllm_dtype (`str`, *optional*, defaults to `"auto"`):
Data type to use for vLLM generation. If set to `"auto"`, the data type will be automatically determined
Data type to use for vLLM generation. If set to `"auto"`, the data type will be automatically determined
based on the model configuration. Find the supported values in the vLLM documentation.
> Parameters that control the training
Expand Down Expand Up @@ -147,15 +147,15 @@ class GRPOConfig(TrainingArguments):
"out-of-memory (OOM) errors during initialization."
},
)

vllm_dtype: Optional[str] = field(
default="auto",
metadata={
"help": "Data type to use for vLLM generation. If set to 'auto', the data type will be automatically "
"determined based on the model configuration. Find the supported values in the vLLM documentation."
},
)

# Parameters that control the training
learning_rate: float = field(
default=1e-6,
Expand Down

0 comments on commit 9684437

Please sign in to comment.