Skip to content

Commit 307f033

Browse files
authored
Drop im2col for VAE since it crashes the driver (huggingface#1010)
This is for untuned models.
1 parent 5167df0 commit 307f033

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

apps/stable_diffusion/src/utils/resources/opt_flags.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,12 +45,12 @@
4545
"untuned": {
4646
"fp16": {
4747
"default_compilation_flags": [
48-
"--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-flow-detach-elementwise-from-named-ops,iree-flow-convert-1x1-filter-conv2d-to-matmul,iree-preprocessing-convert-conv2d-to-img2col,iree-preprocessing-pad-linalg-ops{pad-size=32}))"
48+
"--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-flow-detach-elementwise-from-named-ops,iree-preprocessing-pad-linalg-ops{pad-size=32}))"
4949
]
5050
},
5151
"fp32": {
5252
"default_compilation_flags": [
53-
"--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-flow-detach-elementwise-from-named-ops,iree-flow-convert-1x1-filter-conv2d-to-matmul,iree-preprocessing-convert-conv2d-to-img2col,iree-preprocessing-pad-linalg-ops{pad-size=16}))"
53+
"--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-flow-detach-elementwise-from-named-ops,iree-preprocessing-pad-linalg-ops{pad-size=16}))"
5454
]
5555
}
5656
}

0 commit comments

Comments
 (0)