From 761bfa8364c8c0db24c458f3fa31902c7be95ea2 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Wed, 20 Sep 2023 21:24:16 -0500 Subject: [PATCH] update test for SDXL Unet wrapper --- api/tests/test_diffusers/test_load.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/tests/test_diffusers/test_load.py b/api/tests/test_diffusers/test_load.py index 474b86d9..a7f4c97c 100644 --- a/api/tests/test_diffusers/test_load.py +++ b/api/tests/test_diffusers/test_load.py @@ -123,7 +123,7 @@ class TestPatchPipeline(unittest.TestCase): server = ServerContext() pipeline = MockPipeline() patch_pipeline(server, pipeline, None, ImageParams("test", "txt2img-sdxl", "ddim", "test", 1.0, 10, 1)) - self.assertFalse(isinstance(pipeline.unet, UNetWrapper)) + self.assertTrue(isinstance(pipeline.unet, UNetWrapper)) def test_vae_wrapper(self): server = ServerContext()