From fd3e94450aa9a2b5ab84ef2d35f6b7b73bb81166 Mon Sep 17 00:00:00 2001 From: BuildTools Date: Fri, 13 Jun 2025 07:46:29 -0600 Subject: [PATCH] push local changes, fix docs --- docs/source/en/api/pipelines/chroma.md | 1 + .../transformers/test_models_transformer_chroma.py | 2 +- tests/pipelines/chroma/test_pipeline_chroma.py | 11 ++--------- 3 files changed, 4 insertions(+), 10 deletions(-) diff --git a/docs/source/en/api/pipelines/chroma.md b/docs/source/en/api/pipelines/chroma.md index 0f8c9940f2..22448d88e0 100644 --- a/docs/source/en/api/pipelines/chroma.md +++ b/docs/source/en/api/pipelines/chroma.md @@ -25,6 +25,7 @@ Original model checkpoints for Chroma can be found [here](https://huggingface.co Chroma can use all the same optimizations as Flux. + ## Inference (Single File) diff --git a/tests/models/transformers/test_models_transformer_chroma.py b/tests/models/transformers/test_models_transformer_chroma.py index 5e177cca44..d1a061ce10 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": 192, + "num_attention_heads": 2, "joint_attention_dim": 32, "axes_dims_rope": [4, 4, 8], "approximator_num_channels": 8, diff --git a/tests/pipelines/chroma/test_pipeline_chroma.py b/tests/pipelines/chroma/test_pipeline_chroma.py index ca3c5858ea..b8db37137c 100644 --- a/tests/pipelines/chroma/test_pipeline_chroma.py +++ b/tests/pipelines/chroma/test_pipeline_chroma.py @@ -4,15 +4,8 @@ import numpy as np import torch from transformers import AutoTokenizer, T5EncoderModel -from diffusers import ( - AutoencoderKL, - ChromaPipeline, - ChromaTransformer2DModel, - FlowMatchEulerDiscreteScheduler, -) -from diffusers.utils.testing_utils import ( - torch_device, -) +from diffusers import AutoencoderKL, ChromaPipeline, ChromaTransformer2DModel, FlowMatchEulerDiscreteScheduler +from diffusers.utils.testing_utils import torch_device from ..test_pipelines_common import ( FluxIPAdapterTesterMixin,