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

[WEB] Fix the mlir location of stable-diffusion model (#367)

Browse Source 
Update the location of stable-diffusion mlir file since there is some
problem with iree-compile.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>

Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com>
This commit is contained in:
Gaurav Shukla
2022-09-26 23:26:36 +05:30
committed by GitHub
parent d4eeff0a5d
commit 0228973eef
3 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
web/README.md
View File

@@ -5,7 +5,9 @@ IMPORTER=1 ./setup_venv.sh
source shark.venv/bin/activate
pip install diffusers scipy
cd web
wget -O models_mlir/stable_diffusion.mlir https://storage.googleapis.com/shark_tank/prashant_nod/stable_diff/stable_diff_torch.mlir
wget -O stable_diffusion.mlir https://storage.googleapis.com/shark_tank/prashant_nod/stable_diff/stable_diff_torch.mlir
python index.py
```
This will launch a gradio server with a public URL.
This will launch a gradio server with a public URL like:
Running on public URL: https://xxxxx.gradio.app

2
web/index.py
View File

@@ -84,7 +84,7 @@ with gr.Blocks() as shark_web:
iters = gr.Number(label="Steps", value=2)
mlir_loc = gr.Textbox(
label="Location of MLIR(Relative to SHARK/web/)",
value="./models_mlir/stable_diffusion.mlir",
value="./stable_diffusion.mlir",
)
device = gr.Textbox(label="Device", value="vulkan")
stable_diffusion = gr.Button("Generate image from prompt")

0
web/models_mlir/__init__.py
View File