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

igpu restrictions

Browse Source
This commit is contained in:
Ean Garvey
2024-06-02 12:51:53 -05:00
parent 6abd9ff5cf
commit 6adae49d9b
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
apps/shark_studio/modules/schedulers.py
View File

@@ -101,7 +101,7 @@ def export_scheduler_model(model):
scheduler_model_map = {
"PNDM": export_scheduler_model("PNDMScheduler"),
# "PNDM": export_scheduler_model("PNDMScheduler"),
# "DPMSolverSDE": export_scheduler_model("DpmSolverSDEScheduler"),
"EulerDiscrete": export_scheduler_model("EulerDiscreteScheduler"),
"EulerAncestralDiscrete": export_scheduler_model("EulerAncestralDiscreteScheduler"),

7
apps/shark_studio/web/ui/sd.py
View File

@@ -44,11 +44,12 @@ from apps.shark_studio.web.ui.common_events import lora_changed
from apps.shark_studio.modules import logger
import apps.shark_studio.web.utils.globals as global_obj
# Disabled some models for demo purposes
sd_default_models = [
# "runwayml/stable-diffusion-v1-5",
# "stabilityai/stable-diffusion-2-1-base",
# "stabilityai/stable-diffusion-2-1",
"stabilityai/stable-diffusion-xl-base-1.0",
# "stabilityai/stable-diffusion-xl-base-1.0",
"stabilityai/sdxl-turbo",
]
sd_default_models.extend(get_checkpoints(model_type="scripts"))
@@ -327,6 +328,8 @@ with gr.Blocks(title="Stable Diffusion") as sd_element:
value=512,
step=512,
label="\U00002195\U0000FE0F Height",
interactive=False, # DEMO
visible=False, # DEMO
)
width = gr.Slider(
512,
@@ -334,6 +337,8 @@ with gr.Blocks(title="Stable Diffusion") as sd_element:
value=512,
step=512,
label="\U00002194\U0000FE0F Width",
interactive=False, # DEMO
visible=False, # DEMO
)
with gr.Accordion(
label="\U00002696\U0000FE0F Model Weights",