SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

update examples after ruff being updated (#36972)

Browse Source 
* update

* update

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2025-03-25 18:15:47 +01:00
committed by GitHub
parent a41677a68b
commit 121830ab47
20 changed files with 42 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pytorch/text-generation/run_generation.py
View File

@@ -322,7 +322,7 @@ def main():
parser.add_argument(
"--use_cpu",
action="store_true",
help="Whether or not to use cpu. If set to False, " "we will use gpu/npu or mps device if available",
help="Whether or not to use cpu. If set to False, we will use gpu/npu or mps device if available",
)
parser.add_argument("--num_return_sequences", type=int, default=1, help="The number of samples to generate.")
parser.add_argument(

2
examples/pytorch/text-generation/run_generation_contrastive_search.py
View File

@@ -68,7 +68,7 @@ def main():
parser.add_argument(
"--use_cpu",
action="store_true",
help="Whether or not to use cpu. If set to False, " "we will use gpu/npu or mps device if available",
help="Whether or not to use cpu. If set to False, we will use gpu/npu or mps device if available",
)
parser.add_argument(
"--fp16",