push local changes, fix docs

Author: Ednaordinary

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.
 
+</Tip>
 
 ## Inference (Single File)
 

tests/models/transformers/test_models_transformer_chroma.py

@@ -125,7 +125,7 @@ def prepare_init_args_and_inputs_for_common(self):
             "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,

tests/pipelines/chroma/test_pipeline_chroma.py

@@ -4,15 +4,8 @@
 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,