Skip to content

Commit e3cd69d

Browse files
committed
skip i8 punet test on cpu
1 parent c0be575 commit e3cd69d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

models/turbine_models/tests/sdxl_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -355,8 +355,8 @@ def test05_t2i_generate_images(self):
355355
assert output is not None
356356

357357
def test06_t2i_generate_images_punet(self):
358-
if arguments["device"] in ["vulkan", "cuda"]:
359-
self.skipTest("Have issues with submodels on vulkan, cuda")
358+
if arguments["device"] in ["vulkan", "cuda", "cpu"]:
359+
self.skipTest("Have issues with submodels on vulkan, cuda, cpu")
360360
if getattr(self.pipe, "unet"):
361361
self.pipe.unload_submodel("unet")
362362
self.pipe.use_punet = True

0 commit comments

Comments
 (0)