Skip to content

Commit c0be575

Browse files
committed
Decompose VAE for cpu
1 parent 453fb38 commit c0be575

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

models/turbine_models/tests/sdxl_test.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,11 @@ def setUp(self):
9393
decomp_attn = {
9494
"text_encoder": True,
9595
"unet": False,
96-
"vae": False,
96+
"vae": (
97+
False
98+
if any(x in arguments["device"] for x in ["hip", "rocm"])
99+
else True
100+
),
97101
}
98102
self.pipe = SharkSDPipeline(
99103
arguments["hf_model_name"],

0 commit comments

Comments
 (0)