Skip to content

Fixed vllm and vllm-omni tasks to correct routing logic for HF DLCs#6007

Open
dwarez wants to merge 1 commit into
aws:masterfrom
huggingface:fix-hf-vllm-routing
Open

Fixed vllm and vllm-omni tasks to correct routing logic for HF DLCs#6007
dwarez wants to merge 1 commit into
aws:masterfrom
huggingface:fix-hf-vllm-routing

Conversation

@dwarez

@dwarez dwarez commented Jul 9, 2026

Copy link
Copy Markdown
Contributor

Previously, the set of vLLM and vLLM-omni tasks used in routing for Hugging Face DLCs was wrong, which resulted in the wrong DLCs selection for models like google/gemma-4-E2B-it and other image-text-to-text models.

@dwarez dwarez temporarily deployed to manual-approval July 9, 2026 11:36 — with GitHub Actions Inactive
}

# HuggingFace pipeline tags for models that can generate multiple output modalities.
OMNI_TASKS: Set[str] = {

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

is it always true?
many any-to-any run on vllm, no? https://recipes.vllm.ai/Google/gemma-4-12B-it

@dwarez dwarez temporarily deployed to manual-approval July 9, 2026 14:33 — with GitHub Actions Inactive
Signed-off-by: DWarez <dario.salvati@huggingface.co>
@dwarez dwarez force-pushed the fix-hf-vllm-routing branch from 732d75e to ee08d99 Compare July 10, 2026 07:04
@dwarez dwarez requested a deployment to manual-approval July 10, 2026 07:04 — with GitHub Actions Waiting
@dwarez dwarez requested a deployment to manual-approval July 10, 2026 07:04 — with GitHub Actions Waiting
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