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

[SD][web] Populate checkpoints as dropdown UI (#918)

Browse Source 
Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com>
This commit is contained in:
Gaurav Shukla
2023-02-03 03:29:50 +05:30
committed by GitHub
parent 9b90672f63
commit 339738f8a3
3 changed files with 42 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
apps/stable_diffusion/web/index.py
View File

@@ -1,6 +1,7 @@
import os
import sys
from pathlib import Path
import glob
if "AMD_ENABLE_LLPC" not in os.environ:
os.environ["AMD_ENABLE_LLPC"] = "1"
@@ -57,29 +58,38 @@ with gr.Blocks(title="Stable Diffusion", css=demo_css) as shark_web:
with gr.Row():
with gr.Column(scale=1, min_width=600):
with gr.Row():
with gr.Group():
model_id = gr.Dropdown(
label="Model ID",
value="stabilityai/stable-diffusion-2-1-base",
choices=[
"Linaqruf/anything-v3.0",
"prompthero/openjourney",
"wavymulder/Analog-Diffusion",
"stabilityai/stable-diffusion-2-1",
"stabilityai/stable-diffusion-2-1-base",
"CompVis/stable-diffusion-v1-4",
],
)
custom_model_id = gr.Textbox(
placeholder="check here: https://huggingface.co/models eg. runwayml/stable-diffusion-v1-5",
value="",
label="HuggingFace Model ID",
)
with gr.Group():
ckpt_loc = gr.File(
label="Upload checkpoint",
file_types=[".ckpt", ".safetensors"],
)
model_id = gr.Dropdown(
label="Model ID",
value="stabilityai/stable-diffusion-2-1-base",
choices=[
"Linaqruf/anything-v3.0",
"prompthero/openjourney",
"wavymulder/Analog-Diffusion",
"stabilityai/stable-diffusion-2-1",
"stabilityai/stable-diffusion-2-1-base",
"CompVis/stable-diffusion-v1-4",
],
)
custom_model_id = gr.Textbox(
placeholder="check here: https://huggingface.co/models eg. runwayml/stable-diffusion-v1-5",
value="",
label="HuggingFace Model ID",
)
with gr.Group():
ckpt_path = resource_path("models")
types = (
"*.ckpt",
"*.safetensors",
) # the tuple of file types
ckpt_files = ["None"]
for extn in types:
files = glob.glob(os.path.join(ckpt_path, extn))
ckpt_files.extend(files)
ckpt_loc = gr.Dropdown(
label="Place all checkpoints at SHARK/apps/stable_diffusion/web/models/",
value="None",
choices=ckpt_files,
)
with gr.Group(elem_id="prompt_box_outer"):
prompt = gr.Textbox(