Cleanup sd model map.

nod-ai · Mar 28, 2024 · 0ade2ec · 0ade2ec
1 parent 1827bc3
commit 0ade2ec
Showing 1 changed file with 0 additions and 30 deletions.
diff --git a/apps/shark_studio/api/sd.py b/apps/shark_studio/api/sd.py
@@ -38,42 +38,12 @@
 sd_model_map = {
     "clip": {
         "initializer": clip.export_clip_model,
-        "ireec_flags": [
-            "--iree-flow-collapse-reduction-dims",
-            "--iree-opt-const-expr-hoisting=False",
-            "--iree-codegen-linalg-max-constant-fold-elements=9223372036854775807",
-            "--iree-flow-inline-constants-max-byte-length=0",
-            "--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-preprocessing-pad-linalg-ops{pad-size=16}))",
-        ],
     },
-    # "vae_encode": {
-    #     "initializer": vae.export_vae_model,
-    #     "ireec_flags": [
-    #         "--iree-flow-collapse-reduction-dims",
-    #         "--iree-opt-const-expr-hoisting=False",
-    #         "--iree-codegen-linalg-max-constant-fold-elements=9223372036854775807",
-    #         "--iree-flow-inline-constants-max-byte-length=0",
-    #         "--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-global-opt-detach-elementwise-from-named-ops,iree-global-opt-convert-1x1-filter-conv2d-to-matmul))",
-    #     ],
-    # },
     "unet": {
         "initializer": unet.export_unet_model,
-        "ireec_flags": [
-            "--iree-flow-collapse-reduction-dims",
-            "--iree-opt-const-expr-hoisting=False",
-            "--iree-codegen-linalg-max-constant-fold-elements=9223372036854775807",
-            "--iree-flow-inline-constants-max-byte-length=0",
-            "--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-global-opt-convert-1x1-filter-conv2d-to-matmul,iree-preprocessing-pad-linalg-ops{pad-size=32}))",
-        ],
     },
     "vae_decode": {
         "initializer": vae.export_vae_model,
-        "ireec_flags": [
-            "--iree-opt-const-expr-hoisting=False",
-            "--iree-codegen-linalg-max-constant-fold-elements=9223372036854775807",
-            "--iree-flow-inline-constants-max-byte-length=0",
-            "--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-global-opt-detach-elementwise-from-named-ops,iree-global-opt-convert-1x1-filter-conv2d-to-matmul,iree-preprocessing-pad-linalg-ops{pad-size=32}))",
-        ],
     },
 }