Pass the flags to vae.

2026-01-10 06:17:55 -05:00 · 2022-10-23 11:19:24 -07:00
parent 2741b8be53
commit a48eaaed20
1 changed files with 9 additions and 1 deletions
--- a/shark/examples/shark_inference/stable_diffusion/main.py
+++ b/shark/examples/shark_inference/stable_diffusion/main.py
@@ -48,7 +48,15 @@ def get_models():
        if args.import_mlir == True:
            return get_vae32(), get_unet32_wrapped()
        else:
-            return get_shark_model(GCLOUD_BUCKET, VAE_FP32), get_shark_model(
+            return get_shark_model(
+                GCLOUD_BUCKET,
+                VAE_FP32,
+                [
+                    "--iree-flow-enable-conv-nchw-to-nhwc-transform",
+                    "--iree-flow-enable-padding-linalg-ops",
+                    "--iree-flow-linalg-ops-padding-size=16",
+                ],
+            ), get_shark_model(
                GCLOUD_BUCKET,
                UNET_FP32,
                [