We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e3cd69d commit e3e1dcbCopy full SHA for e3e1dcb
models/turbine_models/tests/sdxl_test.py
@@ -99,6 +99,11 @@ def setUp(self):
99
else True
100
),
101
}
102
+ attn_spec = {
103
+ "text_encoder": None,
104
+ "unet": arguments["attn_spec"],
105
+ "vae": arguments["attn_spec"],
106
+ }
107
self.pipe = SharkSDPipeline(
108
arguments["hf_model_name"],
109
arguments["height"],
@@ -109,7 +114,7 @@ def setUp(self):
114
arguments["device"],
110
115
arguments["iree_target_triple"],
111
116
ireec_flags=None,
112
- attn_spec=arguments["attn_spec"],
117
+ attn_spec=attn_spec,
113
118
decomp_attn=decomp_attn,
119
pipeline_dir="test_vmfbs",
120
external_weights_dir="test_weights",
0 commit comments