Skip to content

Commit 8694f2c

Browse files
EdnaordinaryDN6
authored andcommitted
add encoder test, remove pooled dim
1 parent fd3e944 commit 8694f2c

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
@@ -165,9 +165,3 @@ def test_chroma_image_output_shape(self):
165165
image = pipe(**inputs).images[0]
166166
output_height, output_width, _ = image.shape
167167
assert (output_height, output_width) == (expected_height, expected_width)
168-
169-
@unittest.skip(
170-
"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"
171-
)
172-
def test_encode_prompt_works_in_isolation(self):
173-
pass

0 commit comments

Comments
 (0)