frozenleaves/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-10-20 17:13:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix argument name in benchmarking script (#41086)

Browse Source 
* Fix argument name in benchmarking script

* Adjust vars
This commit is contained in:
Ákos Hadnagy
2025-09-23 13:05:27 +02:00
committed by GitHub
parent 257fe5eea8
commit 62ce6fcb60
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
benchmark_v2/run_benchmarks.py
View File

@ -347,7 +347,7 @@ Examples:
)
parser.add_argument(
"--upload-to-hub",
"--push-to-hub",
type=str,
help="Upload results to HuggingFace Dataset (provide dataset name, e.g., 'username/benchmark-results')",
)
@ -432,7 +432,7 @@ Examples:
# Upload results to HuggingFace Dataset if requested
upload_run_id = None
if args.upload_to_hub:
if args.push_to_hub:
logger.info("=" * 60)
logger.info("UPLOADING TO HUGGINGFACE DATASET")
logger.info("=" * 60)
@ -441,7 +441,7 @@ Examples:
upload_run_id = upload_results_to_hf_dataset(
output_dir=args.output_dir,
summary_file=summary_file,
dataset_name=args.upload_to_hub,
dataset_name=args.push_to_hub,
run_id=effective_run_id,
token=args.token,
logger=logger,
@ -464,12 +464,12 @@ Examples:
logger.info(f"Output directory: {args.output_dir}")
logger.info(f"Summary report: {summary_file}")
if args.upload_to_hub:
if args.push_to_hub:
if upload_run_id:
logger.info(f"HuggingFace Dataset: {args.upload_to_hub}")
logger.info(f"HuggingFace Dataset: {args.push_to_hub}")
logger.info(f"Run ID: {upload_run_id}")
logger.info(
f"View results: https://huggingface.co/datasets/{args.upload_to_hub}/tree/main/{datetime.now().strftime('%Y-%m-%d')}/runs/{upload_run_id}"
f"View results: https://huggingface.co/datasets/{args.push_to_hub}/tree/main/{datetime.now().strftime('%Y-%m-%d')}/runs/{upload_run_id}"
)
else:
logger.warning("Upload to HuggingFace Dataset failed")