From 876649336e3094ef3dd47583d8e7ee9ecee9d76c Mon Sep 17 00:00:00 2001 From: BuildTools Date: Fri, 13 Jun 2025 04:43:31 -0600 Subject: [PATCH] Make most transformer tests work --- tests/models/transformers/test_models_transformer_chroma.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/transformers/test_models_transformer_chroma.py b/tests/models/transformers/test_models_transformer_chroma.py index 8ed7538aaf..a75e7fab47 100644 --- a/tests/models/transformers/test_models_transformer_chroma.py +++ b/tests/models/transformers/test_models_transformer_chroma.py @@ -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], }