LLM/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2025-12-06 12:34:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
fix hunuyanvideo 1.5 offloading tests.
This commit is contained in:
Sayak Paul
2025-12-03 20:37:59 +08:00
committed by GitHub
parent 5ab5946931
commit d96cbacacd
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/models/transformers/test_models_transformer_hunyuan_1_5.py
View File

@@ -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