Skip to content

Commit 06fb995

Browse files
committed
default proj dim
1 parent 16b6e33 commit 06fb995

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/pipelines/test_pipelines_common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -544,7 +544,7 @@ def test_ip_adapter(self, expected_max_diff: float = 1e-4, expected_pipe_slice=N
544544
components = self.get_dummy_components()
545545
pipe = self.pipeline_class(**components).to(torch_device)
546546
pipe.set_progress_bar_config(disable=None)
547-
image_embed_dim = pipe.transformer.config.pooled_projection_dim
547+
image_embed_dim = pipe.transformer.config.pooled_projection_dim or 768
548548

549549
# forward pass without ip adapter
550550
inputs = self._modify_inputs_for_ip_adapter_test(self.get_dummy_inputs(torch_device))

0 commit comments

Comments
 (0)