Skip to content

Commit e3e1dcb

Browse files
committed
Don't use spec for clip by default
1 parent e3cd69d commit e3e1dcb

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

models/turbine_models/tests/sdxl_test.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,6 +99,11 @@ 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+
}
102107
self.pipe = SharkSDPipeline(
103108
arguments["hf_model_name"],
104109
arguments["height"],
@@ -109,7 +114,7 @@ def setUp(self):
109114
arguments["device"],
110115
arguments["iree_target_triple"],
111116
ireec_flags=None,
112-
attn_spec=arguments["attn_spec"],
117+
attn_spec=attn_spec,
113118
decomp_attn=decomp_attn,
114119
pipeline_dir="test_vmfbs",
115120
external_weights_dir="test_weights",

0 commit comments

Comments
 (0)