[Bugfix] Align Hub kernel dependency guidance

Signed-off-by: hsliuustc0106 <liuhongsheng4@huawei.com>
This commit is contained in:
hsliuustc0106 committed 2026-09-03 23:02:19 +00:00
1 parent 6b95de36a6
commit 0527aadf6b
5 files changed
+8 -28

No files matched your search

@@ -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
+5 -5
View File
@@ -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
@@ -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)
@@ -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()
@@ -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