Skip to content

Commit 16b6e33

Browse files
committed
add encoder test, remove pooled dim
1 parent 178c4ec commit 16b6e33

File tree

2 files changed

+0
-7
lines changed

2 files changed

+0
-7
lines changed

src/diffusers/models/transformers/transformer_chroma.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -409,7 +409,6 @@ def __init__(
409409
attention_head_dim: int = 128,
410410
num_attention_heads: int = 24,
411411
joint_attention_dim: int = 4096,
412-
pooled_projection_dim: int = 768,
413412
axes_dims_rope: Tuple[int, ...] = (16, 56, 56),
414413
approximator_num_channels: int = 64,
415414
approximator_hidden_dim: int = 5120,

tests/pipelines/chroma/test_pipeline_chroma.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -168,9 +168,3 @@ def test_chroma_image_output_shape(self):
168168
image = pipe(**inputs).images[0]
169169
output_height, output_width, _ = image.shape
170170
assert (output_height, output_width) == (expected_height, expected_width)
171-
172-
@unittest.skip(
173-
"Chroma uses Flux encode_prompt but uses CFG. This test is incompatible with the pipeline since the test case does not use a negative prompt embeds"
174-
)
175-
def test_encode_prompt_works_in_isolation(self):
176-
pass

0 commit comments

Comments
 (0)