Skip to content

Respect --dataset_max_samples when using --mlcommons_dataset#2223

Merged
regisss merged 1 commit into
huggingface:mainfrom
HabanaAI:dev/pbielak/respect-dataset-max-samples
Aug 24, 2025
Merged

Respect --dataset_max_samples when using --mlcommons_dataset#2223
regisss merged 1 commit into
huggingface:mainfrom
HabanaAI:dev/pbielak/respect-dataset-max-samples

Conversation

@pbielak
Copy link
Copy Markdown
Collaborator

@pbielak pbielak commented Aug 22, 2025

What does this PR do?

The current implementation of run_generation.py does not take into account the value of --dataset_max_samples when using an MLCommons dataset (via the --mlcommons_dataset argument). Limiting the number of dataset samples allows for faster debugging.

The current implementation of `run_generation.py` does not take into
account the value of `--dataset_max_samples` when using an MLCommons
dataset (via the `--mlcommons_dataset` argument). Limiting the number
of dataset samples allows for faster debugging.
@pbielak pbielak requested a review from regisss as a code owner August 22, 2025 14:54
@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

Copy link
Copy Markdown
Collaborator

@regisss regisss left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@regisss regisss merged commit 846fa96 into huggingface:main Aug 24, 2025
2 of 4 checks passed
@pbielak pbielak deleted the dev/pbielak/respect-dataset-max-samples branch August 25, 2025 07:56
imangohari1 pushed a commit to imangohari1/optimum-habana that referenced this pull request Aug 28, 2025
gplutop7 pushed a commit to HabanaAI/optimum-habana-fork that referenced this pull request Oct 15, 2025
…gingface#2223) (huggingface#630)

Co-authored-by: Piotr Bielak <pbielak@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants