Skip to content

Commit d96cbac

Browse files
authored
[tests] fix hunuyanvideo 1.5 offloading tests. (#12782)
fix hunuyanvideo 1.5 offloading tests.
1 parent 5ab5946 commit d96cbac

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/models/transformers/test_models_transformer_hunyuan_1_5.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ class HunyuanVideo15Transformer3DTests(ModelTesterMixin, unittest.TestCase):
2929
model_class = HunyuanVideo15Transformer3DModel
3030
main_input_name = "hidden_states"
3131
uses_custom_attn_processor = True
32+
model_split_percents = [0.99, 0.99, 0.99]
3233

3334
text_embed_dim = 16
3435
text_embed_2_dim = 8

0 commit comments

Comments
 (0)