Move torch_mlir import to the top (#1574)

nod-ai · Jun 22, 2023 · 1b0f391 · 1b0f391
1 parent b9b8955
commit 1b0f391
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/apps/stable_diffusion/web/index.py b/apps/stable_diffusion/web/index.py
@@ -1,6 +1,11 @@
 from multiprocessing import Process, freeze_support
 import os
 import sys
+
+if sys.platform == "darwin":
+    # import before IREE to avoid torch-MLIR library issues
+    import torch_mlir
+
 import shutil
 import PIL, transformers  # ensures inclusion in pysintaller exe generation
 from apps.stable_diffusion.src import args, clear_all