apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

update test for SDXL Unet wrapper

This commit is contained in:
Sean Sube 2023-09-20 21:24:16 -05:00
parent d48dbf7d6e
commit 761bfa8364
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/tests/test_diffusers/test_load.py
View File

@ -123,7 +123,7 @@ class TestPatchPipeline(unittest.TestCase):
server = ServerContext() server = ServerContext()
pipeline = MockPipeline() pipeline = MockPipeline()
patch_pipeline(server, pipeline, None, ImageParams("test", "txt2img-sdxl", "ddim", "test", 1.0, 10, 1)) 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): def test_vae_wrapper(self):
server = ServerContext() server = ServerContext()