Skip to content

Commit f4c91df

Browse files
[WEB] Add pillow dependency (huggingface#453)
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com> Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com>
1 parent 32e1ba8 commit f4c91df

File tree

2 files changed

+16
-13
lines changed

2 files changed

+16
-13
lines changed

setup_venv.ps1

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,6 @@ pip install --pre torch-mlir torch torchvision --extra-index-url https://downloa
3535
pip install --upgrade -f https://nod-ai.github.io/SHARK-Runtime/pip-release-links.html iree-compiler iree-runtime
3636
Write-Host "Building SHARK..."
3737
pip install -e . -f https://llvm.github.io/torch-mlir/package-index/ -f https://nod-ai.github.io/SHARK-Runtime/pip-release-links.html
38-
pip install diffusers transformers scipy gradio
38+
pip install diffusers transformers scipy pillow gradio
3939
Write-Host "Build and installation completed successfully"
4040
Write-Host "Source your venv with ./shark.venv/Scripts/activate"

web/index.py

Lines changed: 15 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
1-
from models.resnet50 import resnet_inf
2-
from models.albert_maskfill import albert_maskfill_inf
1+
# from models.resnet50 import resnet_inf
2+
# from models.albert_maskfill import albert_maskfill_inf
33
from models.stable_diffusion.main import stable_diff_inf
44

5-
# from models.diffusion.v_diffusion import vdiff_inf
5+
# from models.diffusion.v_diffusion import vdiff_inf
66
import gradio as gr
77
from PIL import Image
88

@@ -205,22 +205,25 @@ def debug_event(debug):
205205
384, 768, value=512, step=64, label="Width"
206206
)
207207
with gr.Row():
208-
precision = gr.Radio(
209-
label="Precision",
210-
value="fp32",
211-
choices=["fp16", "fp32"],
208+
scheduler = gr.Radio(
209+
label="Scheduler",
210+
value="LMS",
211+
choices=["PNDM", "LMS", "DDIM"],
212+
interactive=False,
213+
visible=False,
212214
)
213215
device = gr.Radio(
214216
label="Device",
215217
value="vulkan",
216218
choices=["cpu", "cuda", "vulkan"],
219+
interactive=False,
220+
visible=False,
217221
)
218222
with gr.Row():
219-
scheduler = gr.Radio(
220-
label="Scheduler",
221-
value="LMS",
222-
choices=["PNDM", "LMS", "DDIM"],
223-
interactive=False,
223+
precision = gr.Radio(
224+
label="Precision",
225+
value="fp32",
226+
choices=["fp16", "fp32"],
224227
)
225228
seed = gr.Textbox(
226229
value="42", max_lines=1, label="Seed"

0 commit comments

Comments
 (0)