github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-09 13:57:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

[WEB] Add pillow dependency (#453)

Browse Source 
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-11-01 01:27:21 +05:30
committed by GitHub
parent 32e1ba8c0d
commit f4c91df1df
2 changed files with 18 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup_venv.ps1
View File

@@ -35,6 +35,6 @@ pip install --pre torch-mlir torch torchvision --extra-index-url https://downloa
pip install --upgrade -f https://nod-ai.github.io/SHARK-Runtime/pip-release-links.html iree-compiler iree-runtime
Write-Host "Building SHARK..."
pip install -e . -f https://llvm.github.io/torch-mlir/package-index/ -f https://nod-ai.github.io/SHARK-Runtime/pip-release-links.html
pip install diffusers transformers scipy gradio
pip install diffusers transformers scipy pillow gradio
Write-Host "Build and installation completed successfully"
Write-Host "Source your venv with ./shark.venv/Scripts/activate"

31
web/index.py
View File

@@ -1,8 +1,8 @@
from models.resnet50 import resnet_inf
from models.albert_maskfill import albert_maskfill_inf
# from models.resnet50 import resnet_inf
# from models.albert_maskfill import albert_maskfill_inf
from models.stable_diffusion.main import stable_diff_inf
# from models.diffusion.v_diffusion import vdiff_inf
# from models.diffusion.v_diffusion import vdiff_inf
import gradio as gr
from PIL import Image
@@ -204,23 +204,26 @@ with gr.Blocks() as shark_web:
width = gr.Slider(
384, 768, value=512, step=64, label="Width"
)
with gr.Row():
precision = gr.Radio(
label="Precision",
value="fp32",
choices=["fp16", "fp32"],
)
device = gr.Radio(
label="Device",
value="vulkan",
choices=["cpu", "cuda", "vulkan"],
)
with gr.Row():
scheduler = gr.Radio(
label="Scheduler",
value="LMS",
choices=["PNDM", "LMS", "DDIM"],
interactive=False,
visible=False,
)
device = gr.Radio(
label="Device",
value="vulkan",
choices=["cpu", "cuda", "vulkan"],
interactive=False,
visible=False,
)
with gr.Row():
precision = gr.Radio(
label="Precision",
value="fp32",
choices=["fp16", "fp32"],
)
seed = gr.Textbox(
value="42", max_lines=1, label="Seed"