diff --git a/tests/pipelines/test_pipelines_common.py b/tests/pipelines/test_pipelines_common.py index 22c17085a2..978001e8c3 100644 --- a/tests/pipelines/test_pipelines_common.py +++ b/tests/pipelines/test_pipelines_common.py @@ -544,7 +544,7 @@ class FluxIPAdapterTesterMixin: components = self.get_dummy_components() pipe = self.pipeline_class(**components).to(torch_device) pipe.set_progress_bar_config(disable=None) - image_embed_dim = pipe.transformer.config.pooled_projection_dim + image_embed_dim = pipe.transformer.config.pooled_projection_dim or 768 # forward pass without ip adapter inputs = self._modify_inputs_for_ip_adapter_test(self.get_dummy_inputs(torch_device))