Add an option to respect LLPC env var (huggingface#875)

Also add OSX paths

Author: powderluv

shark/examples/shark_inference/stable_diffusion/main.py

@@ -1,6 +1,11 @@
 import os
+import sys
+
+if "AMD_ENABLE_LLPC" not in os.environ:
+    os.environ["AMD_ENABLE_LLPC"] = "1"
 
-os.environ["AMD_ENABLE_LLPC"] = "1"
+if sys.platform == "darwin":
+    os.environ["DYLD_LIBRARY_PATH"] = "/usr/local/lib"
 
 from transformers import CLIPTextModel, CLIPTokenizer
 import torch
@@ -48,7 +53,6 @@
     SharkEulerDiscreteScheduler,
 )
 import time
-import sys
 from shark.iree_utils.compile_utils import dump_isas
 
 # Helper function to profile the vulkan device.

web/index.py

@@ -1,7 +1,13 @@
 import os
+import sys
 from pathlib import Path
 
-os.environ["AMD_ENABLE_LLPC"] = "1"
+if "AMD_ENABLE_LLPC" not in os.environ:
+    os.environ["AMD_ENABLE_LLPC"] = "1"
+
+if sys.platform == "darwin":
+    os.environ["DYLD_LIBRARY_PATH"] = "/usr/local/lib"
+
 import gradio as gr
 from PIL import Image
 from models.stable_diffusion.resources import resource_path, prompt_examples