Skip to content

[tests] skip hunyuan framepack group offloading tests #14551

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
sayakpaul merged 2 commits into from
Aug 22, 2026
Merged

[tests] skip hunyuan framepack group offloading tests #14551

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
6e70f19
skip hunyuan framepack group offloading tests
sayakpaul Aug 21, 2026
dabd2f2
Merge branch 'main' into skip-go-hunyuan-framepack
sayakpaul Aug 22, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions tests/pipelines/hunyuan_video/test_hunyuan_video_framepack.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -389,6 +389,14 @@ def test_sequential_offload_forward_pass_twice(self):
# this test because of MHA (example: HunyuanDiT because of AttentionPooling layer).
pass

@unittest.skip("The image_encoder uses SiglipVisionModel, which does not support group offloading.")
def test_pipeline_level_group_offloading_inference(self):
# Same root cause as the sequential CPU offloading skips above: the attention layer is a
# torch.nn.MultiheadAttention, which passes `self.out_proj.weight` to
# `torch.nn.functional.multi_head_attention_forward` instead of calling `self.out_proj`. The leaf-level
# onload hook on `out_proj` is therefore never triggered and its weights stay on the CPU.
pass

# TODO(aryan): Create a dummy gemma model with smol vocab size
@unittest.skip(
"A very small vocab size is used for fast tests. So, Any kind of prompt other than the empty default used in other tests will lead to a embedding lookup error. This test uses a long prompt that causes the error."
Expand Down
Loading