OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-12-29 19:05:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Basic explanation for custom_code.py template

Browse Source
This commit is contained in:
JashoBell 2022-09-16 14:06:21 -07:00
parent b153ebe3b6
commit 5a797a5612
1 changed files with 21 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
scripts/custom_code.py
View File

@ -4,20 +4,38 @@ import gradio as gr
from modules.processing import Processed
from modules.shared import opts, cmd_opts, state
# An example custom script that allows you to make modifications to how the model object is handled.
# These scripts appear in the lower-left dropdown menu on the txt2img and img2img tabs.
class Script(scripts.Script):
# The title of the script. This is what will be displayed in the dropdown menu.
def title(self):
return "Custom code"
# Determines when the script should be shown in the dropdown menu via the returned value.
# is_img2img is True if the current tab is img2img, and False if it is txt2img. Thus, return
# is_img2img to only show the script on the img2img tab.
def show(self, is_img2img):
return cmd_opts.allow_code
# How the script's is displayed in the UI. See https://gradio.app/docs/#components
# for the different elements you can use and how to specify them.
# Each UI element can return a value, such as a boolean for a checkbox.
# The returned values are passed to the run method as parameters.
def ui(self, is_img2img):
code = gr.Textbox(label="Python code", visible=False, lines=1)
return [code]
# This is where the additional processing is implemented. The parameters include self, the
# model object (a StableDiffusionProcessing class, see processing.py), and the parameters
# returned by the ui method.
# Custom functions can be defined here, and additional libraries can be imported to be used
# in processing. The return value should be a Processed object, which is what is returned
# by the process_images method.
def run(self, p, code):
assert cmd_opts.allow_code, '--allow-code option must be enabled'
@ -37,4 +55,5 @@ class Script(scripts.Script):
exec(compiled, module.__dict__)
return Processed(p, *display_result_data)