AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'dev'

Browse Source
This commit is contained in:
oobabooga 2023-07-30 08:47:50 -07:00
commit c25602eb65
3 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docs/Extensions.md
View File

@ -111,14 +111,16 @@ functions are declared in the same order that they are called at
generation time. generation time.
""" """
import gradio as gr
import torch import torch
from modules import chat from transformers import LogitsProcessor
from modules import chat, shared
from modules.text_generation import ( from modules.text_generation import (
decode, decode,
encode, encode,
generate_reply, generate_reply,
) )
from transformers import LogitsProcessor
params = { params = {
"display_name": "Example Extension", "display_name": "Example Extension",

6
extensions/example/script.py
View File

@ -7,14 +7,16 @@ functions are declared in the same order that they are called at
generation time. generation time.
""" """
import gradio as gr
import torch import torch
from modules import chat from transformers import LogitsProcessor
from modules import chat, shared
from modules.text_generation import ( from modules.text_generation import (
decode, decode,
encode, encode,
generate_reply, generate_reply,
) )
from transformers import LogitsProcessor
params = { params = {
"display_name": "Example Extension", "display_name": "Example Extension",

4
extensions/sd_api_pictures/script.py
View File

@ -332,8 +332,8 @@ def ui():
negative_prompt = gr.Textbox(placeholder=params['negative_prompt'], value=params['negative_prompt'], label='Negative Prompt') negative_prompt = gr.Textbox(placeholder=params['negative_prompt'], value=params['negative_prompt'], label='Negative Prompt')
with gr.Row(): with gr.Row():
with gr.Column(): with gr.Column():
width = gr.Slider(256, 768, value=params['width'], step=64, label='Width') width = gr.Slider(64, 2048, value=params['width'], step=64, label='Width')
height = gr.Slider(256, 768, value=params['height'], step=64, label='Height') height = gr.Slider(64, 2048, value=params['height'], step=64, label='Height')
with gr.Column(variant="compact", elem_id="sampler_col"): with gr.Column(variant="compact", elem_id="sampler_col"):
with gr.Row(elem_id="sampler_row"): with gr.Row(elem_id="sampler_row"):
sampler_name = gr.Dropdown(value=params['sampler_name'], label='Sampling method', elem_id="sampler_box") sampler_name = gr.Dropdown(value=params['sampler_name'], label='Sampling method', elem_id="sampler_box")