Skip to content

GitLab

Unverified Commit d96cbaca authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

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

fix hunuyanvideo 1.5 offloading tests.
parent 5ab59469
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/models/transformers/test_models_transformer_hunyuan_1_5.py
View file @ d96cbaca
...@@ -29,6 +29,7 @@ class HunyuanVideo15Transformer3DTests(ModelTesterMixin, unittest.TestCase): ...@@ -29,6 +29,7 @@ class HunyuanVideo15Transformer3DTests(ModelTesterMixin, unittest.TestCase):
model_class = HunyuanVideo15Transformer3DModel model_class = HunyuanVideo15Transformer3DModel
main_input_name = "hidden_states" main_input_name = "hidden_states"
uses_custom_attn_processor = True uses_custom_attn_processor = True
model_split_percents = [0.99, 0.99, 0.99]
text_embed_dim = 16 text_embed_dim = 16
text_embed_2_dim = 8 text_embed_2_dim = 8
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment