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

push local changes, fix docs

Browse Source
This commit is contained in:
BuildTools
2025-06-13 07:46:29 -06:00
parent 292469d755
commit 178c4ec928
3 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docs/source/en/api/pipelines/chroma.md
View File

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

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": 192,
"num_attention_heads": 2,
"joint_attention_dim": 32,
"axes_dims_rope": [4, 4, 8],
"approximator_num_channels": 8,

11
tests/pipelines/chroma/test_pipeline_chroma.py
View File

@@ -39,14 +39,13 @@ class ChromaPipelineFastTests(
in_channels=4,
num_layers=num_layers,
num_single_layers=num_single_layers,
attention_head_dim=4,
num_attention_heads=4,
attention_head_dim=16,
num_attention_heads=2,
joint_attention_dim=32,
axes_dims_rope=[4, 4, 8],
approximator_in_factor=1,
approximator_hidden_dim=32,
approximator_out_dim=64,
approximator_layers=5,
approximator_num_channels=8,
approximator_hidden_dim=16,
approximator_layers=1,
)
torch.manual_seed(0)