From d96cbacacdb1e0a49ab436c3f647d002e0f411bf Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Wed, 3 Dec 2025 20:37:59 +0800 Subject: [PATCH] [tests] fix hunuyanvideo 1.5 offloading tests. (#12782) fix hunuyanvideo 1.5 offloading tests. --- tests/models/transformers/test_models_transformer_hunyuan_1_5.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/models/transformers/test_models_transformer_hunyuan_1_5.py b/tests/models/transformers/test_models_transformer_hunyuan_1_5.py index 021fcdc9cf..57080bc5b0 100644 --- 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