Skip to content

Commit 56d6ee7

Browse files
authored
Revert change to attention spec handling in sdxl test
1 parent e3e1dcb commit 56d6ee7

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

models/turbine_models/tests/sdxl_test.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -99,11 +99,6 @@ def setUp(self):
9999
else True
100100
),
101101
}
102-
attn_spec = {
103-
"text_encoder": None,
104-
"unet": arguments["attn_spec"],
105-
"vae": arguments["attn_spec"],
106-
}
107102
self.pipe = SharkSDPipeline(
108103
arguments["hf_model_name"],
109104
arguments["height"],
@@ -114,7 +109,7 @@ def setUp(self):
114109
arguments["device"],
115110
arguments["iree_target_triple"],
116111
ireec_flags=None,
117-
attn_spec=attn_spec,
112+
attn_spec=arguments["attn_spec"],
118113
decomp_attn=decomp_attn,
119114
pipeline_dir="test_vmfbs",
120115
external_weights_dir="test_weights",

0 commit comments

Comments
 (0)