Skip to content

Commit 61bb4ef

Browse files
committed
Don't use punet spec.
1 parent e1c4ac2 commit 61bb4ef

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

models/turbine_models/tests/sdxl_test.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -141,8 +141,7 @@ def test00_compile_pipe(self):
141141
self.pipe.use_punet = True
142142
self.pipe.use_i8_punet = True
143143
self.pipe.setup_punet()
144-
if arguments["iree_target_triple"] != "gfx942":
145-
self.pipe.map["unet"]["export_args"]["attn_spec"] = None
144+
self.pipe.map["unet"]["export_args"]["attn_spec"] = None
146145
self.pipe.prepare_all()
147146
self.pipe.load_map()
148147
output = self.pipe.generate_images(

0 commit comments

Comments
 (0)