Don't use punet spec.

nod-ai · Oct 3, 2024 · 61bb4ef · 61bb4ef
1 parent e1c4ac2
commit 61bb4ef
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/models/turbine_models/tests/sdxl_test.py b/models/turbine_models/tests/sdxl_test.py
@@ -141,8 +141,7 @@ def test00_compile_pipe(self):
         self.pipe.use_punet = True
         self.pipe.use_i8_punet = True
         self.pipe.setup_punet()
-        if arguments["iree_target_triple"] != "gfx942":
-            self.pipe.map["unet"]["export_args"]["attn_spec"] = None
+        self.pipe.map["unet"]["export_args"]["attn_spec"] = None
         self.pipe.prepare_all()
         self.pipe.load_map()
         output = self.pipe.generate_images(