Spaces:
Running on Zero
Running on Zero
File size: 2,854 Bytes
619344d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | import random
import gradio as gr
def create_prompt_opts(var_dict, create_negative_prompt=True, prompt='', negatove_prompt=''):
var_dict['prompt'] = gr.Textbox(
prompt, label='Prompt', show_label=False, lines=2, placeholder='Prompt', container=False, interactive=True)
if create_negative_prompt:
var_dict['negative_prompt'] = gr.Textbox(
negatove_prompt, label='Negative prompt', show_label=False, lines=2,
placeholder='Negative prompt', container=False, interactive=True)
def create_generate_bar(var_dict, text='Generate', variant='primary', seed=-1):
with gr.Row(equal_height=False):
var_dict['run_btn'] = gr.Button(text, variant=variant, scale=2)
var_dict['seed'] = gr.Number(
label='Seed', value=seed, min_width=100, precision=0, minimum=-1, maximum=2 ** 31,
elem_classes=['force-hide-container'])
var_dict['random_seed'] = gr.Button('\U0001f3b2\ufe0f', elem_classes=['tool'])
var_dict['reuse_seed'] = gr.Button('\u267b\ufe0f', elem_classes=['tool'])
with gr.Column(visible=False):
var_dict['last_seed'] = gr.Number(value=seed, label='Last seed')
var_dict['reuse_seed'].click(
fn=lambda x: x,
inputs=var_dict['last_seed'],
outputs=var_dict['seed'],
show_progress=False,
api_name=False)
var_dict['random_seed'].click(
fn=lambda: -1,
outputs=var_dict['seed'],
show_progress=False,
api_name=False)
def create_base_opts(var_dict,
steps=24,
min_steps=4,
max_steps=50,
steps_slider_step=1,
guidance_scale=None,
temperature=None,
render=True):
with gr.Column(variant='compact', elem_classes=['custom-spacing'], render=render) as base_opts:
with gr.Row(variant='compact', elem_classes=['force-hide-container']):
var_dict['steps'] = gr.Slider(
min_steps, max_steps, value=steps, step=steps_slider_step, label='Sampling steps',
elem_classes=['force-hide-container'])
with gr.Row(variant='compact', elem_classes=['force-hide-container']):
if guidance_scale is not None:
var_dict['guidance_scale'] = gr.Slider(
0.0, 30.0, value=guidance_scale, step=0.5, label='Guidance scale',
elem_classes=['force-hide-container'])
if temperature is not None:
var_dict['temperature'] = gr.Slider(
0.0, 1.0, value=temperature, step=0.01, label='Temperature',
elem_classes=['force-hide-container'])
return base_opts
def set_seed(seed):
seed = random.randint(0, 2**31) if seed == -1 else seed
return seed
|