riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

More typing in src/transformers/training_args.py (#38106)

Browse Source 
* Annotate `framework` in src/transformers/training_args.py

Signed-off-by: cyy <cyyever@outlook.com>

* Fix typing

Signed-off-by: cyy <cyyever@outlook.com>

* Revert framework change

Signed-off-by: cyy <cyyever@outlook.com>

---------

Signed-off-by: cyy <cyyever@outlook.com>
This commit is contained in:
Yuanyuan Chen 2025-05-22 19:14:33 +08:00 committed by GitHub
parent 03a4c024dc
commit 1234683309
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/transformers/training_args.py
View File

@ -807,8 +807,8 @@ class TrainingArguments:
"gradient_checkpointing_kwargs",
"lr_scheduler_kwargs",
]
framework = "pt"
output_dir: Optional[str] = field(
default=None,
metadata={
@ -920,7 +920,7 @@ class TrainingArguments:
)
warmup_steps: int = field(default=0, metadata={"help": "Linear warmup over warmup_steps."})
log_level: Optional[str] = field(
log_level: str = field(
default="passive",
metadata={
"help": (
@ -931,7 +931,7 @@ class TrainingArguments:
"choices": trainer_log_levels.keys(),
},
)
log_level_replica: Optional[str] = field(
log_level_replica: str = field(
default="warning",
metadata={
"help": "Logger log level to use on replica nodes. Same choices and defaults as ``log_level``",