From 0527aadf6be3e8379b47dce046bf1395bd8f8226 Mon Sep 17 00:00:00 2001 From: hsliuustc0106 Date: Thu, 3 Sep 2026 23:02:19 +0000 Subject: [PATCH] [Bugfix] Align Hub kernel dependency guidance Signed-off-by: hsliuustc0106 --- .../attention_backends/huggingface_hub.md | 2 +- requirements/common.txt | 10 +++++----- tests/config/test_dependency_constraints.py | 20 ------------------- .../models/ltx2/test_ltx25_pipeline.py | 2 +- .../diffusion/models/ltx2/ltx2_components.py | 2 +- 5 files changed, 8 insertions(+), 28 deletions(-) delete mode 100644 tests/config/test_dependency_constraints.py diff --git a/docs/user_guide/diffusion/attention_backends/huggingface_hub.md b/docs/user_guide/diffusion/attention_backends/huggingface_hub.md index 659fe99f7..574ff625a 100644 --- a/docs/user_guide/diffusion/attention_backends/huggingface_hub.md +++ b/docs/user_guide/diffusion/attention_backends/huggingface_hub.md @@ -12,7 +12,7 @@ implementations to reduce numerical drift and sampling divergence. ## Installation ```bash -pip install kernels==0.16.0 +pip install kernels==0.15.2 ``` If `kernels` is unavailable, an explicit `FLASH_ATTN_HUB` or diff --git a/requirements/common.txt b/requirements/common.txt index 908726031..bb79e0a6a 100644 --- a/requirements/common.txt +++ b/requirements/common.txt @@ -1,8 +1,7 @@ # Common dependencies for all platforms # transformers: required unified multimodal model classes are unavailable # before 5.10.1; 5.10.0 is yanked, and 5.15 has a known construction -# regression. kernels 0.16.0's stable-ABI Hub integration also requires -# transformers 5.13.0 or newer. transformers 5.9 also changed +# regression. transformers 5.9 also changed # _adjust_missing_and_unexpected_keys from # list+list to set|set, breaking models that declare # _keys_to_ignore_on_load_unexpected as a list (e.g. OpenMOSS-Team/MOSS-TTS-Nano) @@ -10,12 +9,13 @@ # That breakage is now handled in code instead of by pinning: # MossTTSNanoForGeneration wraps weight loading in # transformers_keys_to_ignore_compat (vllm_omni/model_executor/models/utils.py). -transformers >= 5.13.0, < 5.15 +transformers >= 5.10.1, < 5.15 av>=14.0.0 omegaconf>=2.3.0 diffusers==0.40.0 -# Hub kernels, including LTX-2.5 NATTEN -kernels==0.16.0 +# Hub kernels, including LTX-2.5 NATTEN. Keep this pin in sync with the +# supported transformers range above. +kernels==0.15.2 safetensors>=0.8.0 accelerate==1.12.0 soundfile>=0.13.1 diff --git a/tests/config/test_dependency_constraints.py b/tests/config/test_dependency_constraints.py deleted file mode 100644 index cb7269de5..000000000 --- a/tests/config/test_dependency_constraints.py +++ /dev/null @@ -1,20 +0,0 @@ -# SPDX-License-Identifier: Apache-2.0 -# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project - -from __future__ import annotations - -import re -from pathlib import Path - -import pytest - -pytestmark = [pytest.mark.core_model, pytest.mark.cpu] - -_COMMON_REQUIREMENTS = Path(__file__).parents[2] / "requirements" / "common.txt" - - -def test_common_requirements_keep_hub_kernel_dependencies_compatible(): - requirements = _COMMON_REQUIREMENTS.read_text() - - assert re.search(r"^transformers\s*>=\s*5\.13\.0,\s*<\s*5\.15\s*$", requirements, re.MULTILINE) - assert re.search(r"^kernels\s*==\s*0\.16\.0\s*$", requirements, re.MULTILINE) diff --git a/tests/diffusion/models/ltx2/test_ltx25_pipeline.py b/tests/diffusion/models/ltx2/test_ltx25_pipeline.py index 4b1fef7d5..950906190 100644 --- a/tests/diffusion/models/ltx2/test_ltx25_pipeline.py +++ b/tests/diffusion/models/ltx2/test_ltx25_pipeline.py @@ -386,7 +386,7 @@ def test_ltx25_natten_failure_has_actionable_omni_remedy(monkeypatch): MissingNattenProcessor, ) - with pytest.raises(RuntimeError, match=r"kernels==0\.14\.1.*supported GPU.*allow Hub access"): + with pytest.raises(RuntimeError, match=r"kernels==0\.15\.2.*supported GPU.*allow Hub access"): ltx2_components._create_ltx25_natten_processor() diff --git a/vllm_omni/diffusion/models/ltx2/ltx2_components.py b/vllm_omni/diffusion/models/ltx2/ltx2_components.py index 6325df870..a99574e36 100644 --- a/vllm_omni/diffusion/models/ltx2/ltx2_components.py +++ b/vllm_omni/diffusion/models/ltx2/ltx2_components.py @@ -298,7 +298,7 @@ def _create_ltx25_natten_processor() -> LTX2VideoVaeNeighborhoodNattenProcessor: except (ImportError, OSError, RuntimeError, ValueError) as exc: raise RuntimeError( "LTX-2.5 DiffVAE requires the shi-labs/natten Hub kernel. " - "Install kernels==0.14.1, use a supported GPU, leave " + "Install kernels==0.15.2, use a supported GPU, leave " "DIFFUSERS_DISABLE_REMOTE_CODE unset, and allow Hub access during kernel initialization." ) from exc