AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

Make most transformer tests work

Browse Source
This commit is contained in:
BuildTools
2025-06-13 04:43:31 -06:00
parent 829c6f199e
commit 876649336e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/transformers/test_models_transformer_chroma.py
View File

@@ -125,7 +125,7 @@ class ChromaTransformerTests(ModelTesterMixin, unittest.TestCase):
"num_layers": 1,
"num_single_layers": 1,
"attention_head_dim": 16,
"num_attention_heads": 2,
"num_attention_heads": 192,
"joint_attention_dim": 32,
"axes_dims_rope": [4, 4, 8],
}