Skip to content

accept any jobs hardware flavor - #278

Merged
evalstate merged 1 commit into
huggingface:mainfrom
pngwn:jobs-flavor-string
Oct 19, 2026
Merged

evalstate merged 1 commit into
huggingface:mainfrom
pngwn:jobs-flavor-string

Conversation

@pngwn

@pngwn pngwn commented Oct 19, 2026

Copy link
Copy Markdown
Member

hf_jobs now accepts any hardware flavor and leaves validation to the Jobs API. Before this, the flavor was a fixed list, so current hardware like h200 and rtx-pro-6000 was rejected before the job reached the Hub:

{ "operation": "run", "args": { "image": "python:3.12", "command": ["nvidia-smi"], "flavor": "h200" } }
before: Invalid option: expected one of "cpu-basic"|"cpu-upgrade"|...|"a100x8"|"inf2x6"
after:  the job is submitted on h200

The known flavors are still listed in the param description and the help text as hints. They now match GET /api/jobs/hardware, which adds the h200 and rtx-pro-6000 sizes and drops zero-a10g, sprx8 and inf2x6 (the Jobs API doesn't list those). A misspelled or unavailable flavor now comes back as the Jobs API's own error. New hardware no longer needs a schema change or an @huggingface/hub bump.

@evalstate
evalstate merged commit a6824cb into huggingface:main Oct 19, 2026
3 checks passed
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.

2 participants