ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge branch 'video-wan' into video-hunyuan-gate

Browse source
This commit is contained in:
Daniel Han 2026-07-05 02:32:31 +00:00
commit 5b92ee67cd
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
studio/backend/tests/test_video_backend.py
View file

@ -814,8 +814,13 @@ def test_wan_a14b_step_cache_applies_to_both_dits(fake_runtime):
assert status["transformer_cache"] == "fbcache"
def test_wan_a14b_attention_applies_to_both_dits(fake_runtime):
# An explicit attention backend must be set on both experts.
def test_wan_a14b_attention_applies_to_both_dits(fake_runtime, monkeypatch):
# An explicit attention backend must be set on both experts. The fake runtime is a
# CPU target, where the NVIDIA gate correctly drops explicit kernels; pin the gate
# open so the explicit-set path itself is what this test exercises.
from core.inference import diffusion_attention as attn_mod
monkeypatch.setattr(attn_mod, "_is_cuda_nvidia", lambda target: True)
backend = VideoBackend()
backend.load_pipeline(
"Wan-AI/Wan2.2-T2V-A14B-Diffusers",