github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
Also add OSX paths
This commit is contained in:
powderluv
2023-01-25 13:56:55 -08:00
committed by GitHub
parent c3a641f0ab
commit 9bbffa519e
2 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
shark/examples/shark_inference/stable_diffusion/main.py
View File

@@ -1,6 +1,11 @@
import os
import sys
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"
from transformers import CLIPTextModel, CLIPTokenizer
import torch
@@ -48,7 +53,6 @@ from schedulers import (
SharkEulerDiscreteScheduler,
)
import time
import sys
from shark.iree_utils.compile_utils import dump_isas
# Helper function to profile the vulkan device.

8
web/index.py
View File

@@ -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