Fixed tokenizer being broken in some environments
Imrayya
parent
33683c21ea
commit
bcc7a3ee53
|
|
@ -31,52 +31,6 @@ def get_list_blacklist():
|
||||||
|
|
||||||
|
|
||||||
def on_ui_tabs():
|
def on_ui_tabs():
|
||||||
# structure
|
|
||||||
|
|
||||||
txt2img_prompt = modules.ui.txt2img_paste_fields[0][0]
|
|
||||||
img2img_prompt = modules.ui.img2img_paste_fields[0][0]
|
|
||||||
|
|
||||||
with gr.Blocks(analytics_enabled=False) as prompt_generator:
|
|
||||||
with gr.Column():
|
|
||||||
with gr.Row():
|
|
||||||
promptTxt = gr.Textbox(
|
|
||||||
lines=2, elem_id="promptTxt", label="Start of the prompt")
|
|
||||||
with gr.Column():
|
|
||||||
with gr.Row():
|
|
||||||
temp_slider = gr.Slider(
|
|
||||||
elem_id="temp_slider", label="Temperature", interactive=True, minimum=0, maximum=1, value=0.9)
|
|
||||||
max_length_slider = gr.Slider(
|
|
||||||
elem_id="max_length_slider", label="Max Length", interactive=True, minimum=1, maximum=200, step=1, value=80)
|
|
||||||
top_k_slider = gr.Slider(
|
|
||||||
elem_id="top_k_slider", label="Top K", value=8, minimum=1, maximum=20, interactive=True)
|
|
||||||
with gr.Column():
|
|
||||||
with gr.Row():
|
|
||||||
repetition_penalty_slider = gr.Slider(
|
|
||||||
elem_id="repetition_penalty_slider", label="Repetition Penalty", value=1.2, minimum=0, maximum=10, interactive=True)
|
|
||||||
num_return_sequences_slider = gr.Slider(
|
|
||||||
elem_id="num_return_sequences_slider", label="How Many To Generate", value=5, minimum=1, maximum=20, interactive=True, step=1)
|
|
||||||
with gr.Column():
|
|
||||||
with gr.Row():
|
|
||||||
use_blacklist_checkbox = gr.Checkbox(label="Use blacklist?")
|
|
||||||
gr.HTML(value="<center>Using <code>\".\extensions\stable-diffusion-webui-Prompt_Generator\\blacklist.txt</code>\".<br>It will delete any matches to the generated result (case insensitive).</center>")
|
|
||||||
with gr.Column():
|
|
||||||
with gr.Row():
|
|
||||||
generateButton = gr.Button(
|
|
||||||
value="Generate", elem_id="generate_button")
|
|
||||||
with gr.Column(visible=False) as results_col:
|
|
||||||
results = gr.Text(
|
|
||||||
label="Results", elem_id="Results_textBox", interactive=False)
|
|
||||||
with gr.Column(visible=False) as promptNum_col:
|
|
||||||
with gr.Row():
|
|
||||||
promptNum = gr.Textbox(
|
|
||||||
lines=1, elem_id="promptNum", label="Send which prompt")
|
|
||||||
with gr.Column():
|
|
||||||
warning = gr.HTML(
|
|
||||||
value="Select one number and send that prompt to txt2img or img2img", visible=False)
|
|
||||||
with gr.Row():
|
|
||||||
send_to_txt2img = gr.Button('Send to txt2img', visible=False)
|
|
||||||
send_to_img2img = gr.Button('Send to img2img', visible=False)
|
|
||||||
|
|
||||||
# Method to create the extended prompt
|
# Method to create the extended prompt
|
||||||
|
|
||||||
def generate_longer_prompt(prompt, temperature, top_k,
|
def generate_longer_prompt(prompt, temperature, top_k,
|
||||||
|
|
@ -129,6 +83,53 @@ def on_ui_tabs():
|
||||||
f"Exception encountered while attempting to generate prompt: {e}")
|
f"Exception encountered while attempting to generate prompt: {e}")
|
||||||
return gr.update(), f"Error: {e}"
|
return gr.update(), f"Error: {e}"
|
||||||
|
|
||||||
|
|
||||||
|
# structure
|
||||||
|
txt2img_prompt = modules.ui.txt2img_paste_fields[0][0]
|
||||||
|
img2img_prompt = modules.ui.img2img_paste_fields[0][0]
|
||||||
|
|
||||||
|
with gr.Blocks(analytics_enabled=False) as prompt_generator:
|
||||||
|
with gr.Column():
|
||||||
|
with gr.Row():
|
||||||
|
promptTxt = gr.Textbox(
|
||||||
|
lines=2, elem_id="promptTxt", label="Start of the prompt")
|
||||||
|
with gr.Column():
|
||||||
|
with gr.Row():
|
||||||
|
temp_slider = gr.Slider(
|
||||||
|
elem_id="temp_slider", label="Temperature", interactive=True, minimum=0, maximum=1, value=0.9)
|
||||||
|
max_length_slider = gr.Slider(
|
||||||
|
elem_id="max_length_slider", label="Max Length", interactive=True, minimum=1, maximum=200, step=1, value=80)
|
||||||
|
top_k_slider = gr.Slider(
|
||||||
|
elem_id="top_k_slider", label="Top K", value=8, minimum=1, maximum=20, interactive=True)
|
||||||
|
with gr.Column():
|
||||||
|
with gr.Row():
|
||||||
|
repetition_penalty_slider = gr.Slider(
|
||||||
|
elem_id="repetition_penalty_slider", label="Repetition Penalty", value=1.2, minimum=0, maximum=10, interactive=True)
|
||||||
|
num_return_sequences_slider = gr.Slider(
|
||||||
|
elem_id="num_return_sequences_slider", label="How Many To Generate", value=5, minimum=1, maximum=20, interactive=True, step=1)
|
||||||
|
with gr.Column():
|
||||||
|
with gr.Row():
|
||||||
|
use_blacklist_checkbox = gr.Checkbox(label="Use blacklist?")
|
||||||
|
gr.HTML(value="<center>Using <code>\".\extensions\stable-diffusion-webui-Prompt_Generator\\blacklist.txt</code>\".<br>It will delete any matches to the generated result (case insensitive).</center>")
|
||||||
|
with gr.Column():
|
||||||
|
with gr.Row():
|
||||||
|
generateButton = gr.Button(
|
||||||
|
value="Generate", elem_id="generate_button")
|
||||||
|
with gr.Column(visible=False) as results_col:
|
||||||
|
results = gr.Text(
|
||||||
|
label="Results", elem_id="Results_textBox", interactive=False)
|
||||||
|
with gr.Column(visible=False) as promptNum_col:
|
||||||
|
with gr.Row():
|
||||||
|
promptNum = gr.Textbox(
|
||||||
|
lines=1, elem_id="promptNum", label="Send which prompt")
|
||||||
|
with gr.Column():
|
||||||
|
warning = gr.HTML(
|
||||||
|
value="Select one number and send that prompt to txt2img or img2img", visible=False)
|
||||||
|
with gr.Row():
|
||||||
|
send_to_txt2img = gr.Button('Send to txt2img', visible=False)
|
||||||
|
send_to_img2img = gr.Button('Send to img2img', visible=False)
|
||||||
|
|
||||||
|
|
||||||
# events
|
# events
|
||||||
generateButton.click(fn=generate_longer_prompt, inputs=[
|
generateButton.click(fn=generate_longer_prompt, inputs=[
|
||||||
promptTxt, temp_slider, top_k_slider, max_length_slider,
|
promptTxt, temp_slider, top_k_slider, max_length_slider,
|
||||||
|
|
@ -146,5 +147,4 @@ def on_ui_tabs():
|
||||||
inputs=None, outputs=None)
|
inputs=None, outputs=None)
|
||||||
return (prompt_generator, "Prompt Generator", "Prompt Generator"),
|
return (prompt_generator, "Prompt Generator", "Prompt Generator"),
|
||||||
|
|
||||||
|
|
||||||
script_callbacks.on_ui_tabs(on_ui_tabs)
|
script_callbacks.on_ui_tabs(on_ui_tabs)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue