AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-19 15:56:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Get SD samplers from API (#2889)

Browse Source
This commit is contained in:
missionfloyd 2023-06-27 14:31:54 -06:00 committed by GitHub
parent d7a7f7896b
commit cb029cf65f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 45 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
extensions/sd_api_pictures/script.py
View File

@ -12,6 +12,7 @@ from PIL import Image
import modules.shared as shared
from modules.models import reload_model, unload_model
from modules.ui import create_refresh_button
torch._C._jit_set_profiling_mode(False)
@ -76,7 +77,6 @@ def give_VRAM_priority(actor):
if params['manage_VRAM']:
give_VRAM_priority('set')
samplers = ['DDIM', 'DPM++ 2M Karras'] # TODO: get the availible samplers with http://{address}}/sdapi/v1/samplers
SD_models = ['NeverEndingDream'] # TODO: get with http://{address}}/sdapi/v1/sd-models and allow user to select
picture_response = False # specifies if the next model response should appear as a picture
@ -248,7 +248,6 @@ def filter_address(address):
def SD_api_address_update(address):
global params
msg = "✔️ SD API is found on:"
@ -291,6 +290,17 @@ def load_checkpoint(checkpoint):
requests.post(url=f'{params["address"]}/sdapi/v1/options', json=payload)
def get_samplers():
try:
response = requests.get(url=f'{params["address"]}/sdapi/v1/samplers')
response.raise_for_status()
samplers = [x["name"] for x in response.json()]
except:
samplers = []
return samplers
def ui():
# Gradio elements
@ -317,9 +327,11 @@ def ui():
with gr.Column():
width = gr.Slider(256, 768, value=params['width'], step=64, label='Width')
height = gr.Slider(256, 768, value=params['height'], step=64, label='Height')
with gr.Column():
sampler_name = gr.Textbox(placeholder=params['sampler_name'], value=params['sampler_name'], label='Sampling method', elem_id="sampler_box")
steps = gr.Slider(1, 150, value=params['steps'], step=1, label="Sampling steps")
with gr.Column(variant="compact", elem_id="sampler_col"):
with gr.Row(elem_id="sampler_row"):
sampler_name = gr.Dropdown(value=params['sampler_name'], label='Sampling method', elem_id="sampler_box")
create_refresh_button(sampler_name, lambda: None, lambda: {'choices': get_samplers()}, 'refresh-button')
steps = gr.Slider(1, 150, value=params['steps'], step=1, label="Sampling steps", elem_id="steps_box")
with gr.Row():
seed = gr.Number(label="Seed", value=params['seed'], elem_id="seed_box")
cfg_scale = gr.Number(label="CFG Scale", value=params['cfg_scale'], elem_id="cfg_box")

29
extensions/sd_api_pictures/style.css
View File

@ -2,6 +2,32 @@
.SDAP #sampler_box {
padding-top: var(--spacing-sm);
padding-bottom: var(--spacing-sm);
border: 0;
}
.SDAP #steps_box {
border-radius: 0 0 var(--block-radius) var(--block-radius);
}
.SDAP #sampler_col {
gap: 0;
padding: 0;
background-color: transparent;
}
.SDAP #sampler_row {
border-bottom: 0;
box-shadow: var(--block-shadow);
border-width: var(--block-border-width);
border-color: var(--block-border-color);
border-radius: var(--block-radius) var(--block-radius) 0 0;
background: var(--block-background-fill);
gap: 0;
}
.SDAP #sampler_row .refresh-button {
margin-bottom: var(--spacing-sm);
margin-right: var(--spacing-lg);
}
.SDAP #seed_box,
@ -11,7 +37,8 @@
.SDAP #sampler_box span,
.SDAP #seed_box span,
.SDAP #cfg_box span{
.SDAP #cfg_box span,
.SDAP #steps_box span {
margin-bottom: var(--spacing-sm);
}