[tests] fix hunuyanvideo 1.5 offloading tests. (#12782)

fix hunuyanvideo 1.5 offloading tests.

[tests] fix hunuyanvideo 1.5 offloading tests. (#12782)
fix hunuyanvideo 1.5 offloading tests.
d96cbaca · Sayak Paul · GitHub · 5ab59469 · d96cbaca
Unverified Commit d96cbaca authored Dec 03, 2025 by Sayak Paul Committed by GitHub Dec 03, 2025
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

tests/models/transformers/test_models_transformer_hunyuan_1_5.py ...odels/transformers/test_models_transformer_hunyuan_1_5.py +1 -0

No files found.
--- a/tests/models/transformers/test_models_transformer_hunyuan_1_5.py
+++ b/tests/models/transformers/test_models_transformer_hunyuan_1_5.py
@@ -29,6 +29,7 @@ class HunyuanVideo15Transformer3DTests(ModelTesterMixin, unittest.TestCase):
    model_class = HunyuanVideo15Transformer3DModel
    main_input_name = "hidden_states"
    uses_custom_attn_processor = True
+    model_split_percents = [0.99, 0.99, 0.99]

    text_embed_dim = 16
    text_embed_2_dim = 8