Skip to content

Commit 7ed8eab

Browse files
committed
fix tests
1 parent a1b5c1e commit 7ed8eab

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

tests/pipelines/stable_diffusion/test_stable_diffusion_adapter.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -339,8 +339,8 @@ def test_stable_diffusion_adapter_default_case(self):
339339

340340

341341
class StableDiffusionLightAdapterPipelineFastTests(AdapterTests, PipelineTesterMixin, unittest.TestCase):
342-
def get_dummy_components(self):
343-
return super().get_dummy_components("light_adapter")
342+
def get_dummy_components(self, time_cond_proj_dim=None):
343+
return super().get_dummy_components("light_adapter", time_cond_proj_dim=time_cond_proj_dim)
344344

345345
def get_dummy_components_with_full_downscaling(self):
346346
return super().get_dummy_components_with_full_downscaling("light_adapter")
@@ -362,8 +362,8 @@ def test_stable_diffusion_adapter_default_case(self):
362362

363363

364364
class StableDiffusionMultiAdapterPipelineFastTests(AdapterTests, PipelineTesterMixin, unittest.TestCase):
365-
def get_dummy_components(self):
366-
return super().get_dummy_components("multi_adapter")
365+
def get_dummy_components(self, time_cond_proj_dim=None):
366+
return super().get_dummy_components("multi_adapter", time_cond_proj_dim=time_cond_proj_dim)
367367

368368
def get_dummy_components_with_full_downscaling(self):
369369
return super().get_dummy_components_with_full_downscaling("multi_adapter")

tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_adapter.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -393,8 +393,8 @@ def test_adapter_sdxl_lcm(self):
393393
class StableDiffusionXLMultiAdapterPipelineFastTests(
394394
StableDiffusionXLAdapterPipelineFastTests, PipelineTesterMixin, unittest.TestCase
395395
):
396-
def get_dummy_components(self):
397-
return super().get_dummy_components("multi_adapter")
396+
def get_dummy_components(self, time_cond_proj_dim=None):
397+
return super().get_dummy_components("multi_adapter", time_cond_proj_dim=time_cond_proj_dim)
398398

399399
def get_dummy_components_with_full_downscaling(self):
400400
return super().get_dummy_components_with_full_downscaling("multi_adapter")

0 commit comments

Comments
 (0)