2022-09-03 12:08:45 +03:00
import argparse
2022-09-30 01:46:23 +03:00
import datetime
2022-09-03 12:08:45 +03:00
import json
import os
2022-09-30 01:46:23 +03:00
import sys
2022-09-03 12:08:45 +03:00
import gradio as gr
2022-09-08 16:37:13 +03:00
import tqdm
2022-09-03 12:08:45 +03:00
2022-09-05 23:08:06 +03:00
import modules . artists
2022-09-11 18:48:36 +03:00
import modules . interrogate
2022-09-17 07:49:31 +03:00
import modules . memmon
2022-09-17 12:05:04 +03:00
import modules . sd_models
2022-09-30 01:46:23 +03:00
import modules . styles
2022-10-04 11:24:35 +03:00
import modules . devices as devices
2022-10-13 19:22:41 +03:00
from modules import sd_samplers , sd_models
2022-10-11 15:51:22 +03:00
from modules . hypernetworks import hypernetwork
2022-10-06 18:19:52 +03:00
from modules . paths import models_path , script_path , sd_path
2022-09-03 12:08:45 +03:00
sd_model_file = os . path . join ( script_path , ' model.ckpt ' )
2022-09-17 12:05:04 +03:00
default_sd_model_file = sd_model_file
2022-09-03 12:08:45 +03:00
parser = argparse . ArgumentParser ( )
parser . add_argument ( " --config " , type = str , default = os . path . join ( sd_path , " configs/stable-diffusion/v1-inference.yaml " ) , help = " path to config which constructs model " , )
2022-09-30 22:16:03 +03:00
parser . add_argument ( " --ckpt " , type = str , default = sd_model_file , help = " path to checkpoint of stable diffusion model; if specified, this checkpoint will be added to the list of checkpoints and loaded " , )
2022-09-30 11:42:40 +03:00
parser . add_argument ( " --ckpt-dir " , type = str , default = None , help = " Path to directory with stable diffusion checkpoints " )
2022-09-03 12:08:45 +03:00
parser . add_argument ( " --gfpgan-dir " , type = str , help = " GFPGAN directory " , default = ( ' ./src/gfpgan ' if os . path . exists ( ' ./src/gfpgan ' ) else ' ./GFPGAN ' ) )
2022-09-23 10:26:00 +03:00
parser . add_argument ( " --gfpgan-model " , type = str , help = " GFPGAN model file name " , default = None )
2022-09-03 12:08:45 +03:00
parser . add_argument ( " --no-half " , action = ' store_true ' , help = " do not switch the model to 16-bit floats " )
2022-10-10 16:11:14 +03:00
parser . add_argument ( " --no-half-vae " , action = ' store_true ' , help = " do not switch the VAE model to 16-bit floats " )
2022-09-15 06:04:05 +03:00
parser . add_argument ( " --no-progressbar-hiding " , action = ' store_true ' , help = " do not hide progressbar in gradio UI (we hide it because it slows down ML if you have hardware acceleration in browser) " )
2022-09-03 12:08:45 +03:00
parser . add_argument ( " --max-batch-count " , type = int , default = 16 , help = " maximum batch count value for the UI " )
2022-09-13 09:42:51 +03:00
parser . add_argument ( " --embeddings-dir " , type = str , default = os . path . join ( script_path , ' embeddings ' ) , help = " embeddings directory for textual inversion (default: embeddings) " )
2022-10-11 14:53:02 +03:00
parser . add_argument ( " --hypernetwork-dir " , type = str , default = os . path . join ( models_path , ' hypernetworks ' ) , help = " hypernetwork directory " )
2022-09-03 12:08:45 +03:00
parser . add_argument ( " --allow-code " , action = ' store_true ' , help = " allow custom script execution from webui " )
2022-09-09 02:17:38 +03:00
parser . add_argument ( " --medvram " , action = ' store_true ' , help = " enable stable diffusion model optimizations for sacrificing a little speed for low VRM usage " )
parser . add_argument ( " --lowvram " , action = ' store_true ' , help = " enable stable diffusion model optimizations for sacrificing a lot of speed for very low VRM usage " )
2022-09-18 11:14:42 +03:00
parser . add_argument ( " --always-batch-cond-uncond " , action = ' store_true ' , help = " disables cond/uncond batching that is enabled to save memory with --medvram or --lowvram " )
2022-09-12 18:40:06 +03:00
parser . add_argument ( " --unload-gfpgan " , action = ' store_true ' , help = " does not do anything. " )
2022-09-03 12:08:45 +03:00
parser . add_argument ( " --precision " , type = str , help = " evaluate at this precision " , choices = [ " full " , " autocast " ] , default = " autocast " )
parser . add_argument ( " --share " , action = ' store_true ' , help = " use share=True for gradio and make the UI accessible through their site (doesn ' t work for me but you might have better luck) " )
2022-10-11 12:40:27 +03:00
parser . add_argument ( " --ngrok " , type = str , help = " ngrok authtoken, alternative to gradio --share " , default = None )
2022-10-06 18:19:52 +03:00
parser . add_argument ( " --codeformer-models-path " , type = str , help = " Path to directory with codeformer model file(s). " , default = os . path . join ( models_path , ' Codeformer ' ) )
parser . add_argument ( " --gfpgan-models-path " , type = str , help = " Path to directory with GFPGAN model file(s). " , default = os . path . join ( models_path , ' GFPGAN ' ) )
parser . add_argument ( " --esrgan-models-path " , type = str , help = " Path to directory with ESRGAN model file(s). " , default = os . path . join ( models_path , ' ESRGAN ' ) )
parser . add_argument ( " --bsrgan-models-path " , type = str , help = " Path to directory with BSRGAN model file(s). " , default = os . path . join ( models_path , ' BSRGAN ' ) )
parser . add_argument ( " --realesrgan-models-path " , type = str , help = " Path to directory with RealESRGAN model file(s). " , default = os . path . join ( models_path , ' RealESRGAN ' ) )
parser . add_argument ( " --scunet-models-path " , type = str , help = " Path to directory with ScuNET model file(s). " , default = os . path . join ( models_path , ' ScuNET ' ) )
parser . add_argument ( " --swinir-models-path " , type = str , help = " Path to directory with SwinIR model file(s). " , default = os . path . join ( models_path , ' SwinIR ' ) )
parser . add_argument ( " --ldsr-models-path " , type = str , help = " Path to directory with LDSR model file(s). " , default = os . path . join ( models_path , ' LDSR ' ) )
2022-10-08 17:02:18 +03:00
parser . add_argument ( " --xformers " , action = ' store_true ' , help = " enable xformers for cross attention layers " )
2022-10-08 19:22:15 +03:00
parser . add_argument ( " --force-enable-xformers " , action = ' store_true ' , help = " enable xformers for cross attention layers regardless of whether the checking code thinks you can run it; do not make bug reports if this fails to work " )
2022-10-08 19:02:56 +03:00
parser . add_argument ( " --deepdanbooru " , action = ' store_true ' , help = " enable deepdanbooru interrogator " )
2022-10-11 05:48:54 +03:00
parser . add_argument ( " --opt-split-attention " , action = ' store_true ' , help = " force-enables Doggettx ' s cross-attention layer optimization. By default, it ' s on for torch cuda. " )
parser . add_argument ( " --opt-split-attention-invokeai " , action = ' store_true ' , help = " force-enables InvokeAI ' s cross-attention layer optimization. By default, it ' s on when cuda is unavailable. " )
2022-09-19 00:39:41 +03:00
parser . add_argument ( " --opt-split-attention-v1 " , action = ' store_true ' , help = " enable older version of split attention optimization that does not consume all the VRAM it can find " )
2022-10-11 05:48:54 +03:00
parser . add_argument ( " --disable-opt-split-attention " , action = ' store_true ' , help = " force-disables cross-attention layer optimization " )
2022-10-04 12:22:50 +03:00
parser . add_argument ( " --use-cpu " , nargs = ' + ' , choices = [ ' SD ' , ' GFPGAN ' , ' BSRGAN ' , ' ESRGAN ' , ' SCUNet ' , ' CodeFormer ' ] , help = " use CPU as torch device for specified modules " , default = [ ] )
2022-09-05 19:37:11 +03:00
parser . add_argument ( " --listen " , action = ' store_true ' , help = " launch gradio with 0.0.0.0 as server name, allowing to respond to network requests " )
2022-09-08 11:42:21 +03:00
parser . add_argument ( " --port " , type = int , help = " launch gradio with given server port, you need root/admin rights for ports < 1024, defaults to 7860 if available " , default = None )
2022-09-10 15:41:29 +03:00
parser . add_argument ( " --show-negative-prompt " , action = ' store_true ' , help = " does not do anything " , default = False )
2022-09-10 08:18:54 +03:00
parser . add_argument ( " --ui-config-file " , type = str , help = " filename to use for ui configuration " , default = os . path . join ( script_path , ' ui-config.json ' ) )
2022-09-11 23:00:42 +03:00
parser . add_argument ( " --hide-ui-dir-config " , action = ' store_true ' , help = " hide directory configuration from webui " , default = False )
2022-09-12 09:44:45 +03:00
parser . add_argument ( " --ui-settings-file " , type = str , help = " filename to use for ui settings " , default = os . path . join ( script_path , ' config.json ' ) )
2022-09-12 12:40:55 +03:00
parser . add_argument ( " --gradio-debug " , action = ' store_true ' , help = " launch gradio with --debug option " )
2022-09-12 15:19:14 +03:00
parser . add_argument ( " --gradio-auth " , type = str , help = ' set gradio authentication like " username:password " ; or comma-delimit multiple like " u1:p1,u2:p2,u3:p3 " ' , default = None )
2022-10-06 05:11:15 +03:00
parser . add_argument ( " --gradio-img2img-tool " , type = str , help = ' gradio image uploader tool: can be either editor for ctopping, or color-sketch for drawing ' , choices = [ " color-sketch " , " editor " ] , default = " editor " )
2022-09-12 16:43:11 +03:00
parser . add_argument ( " --opt-channelslast " , action = ' store_true ' , help = " change memory type for stable diffusion to channels last " )
2022-09-14 01:18:07 +03:00
parser . add_argument ( " --styles-file " , type = str , help = " filename to use for styles " , default = os . path . join ( script_path , ' styles.csv ' ) )
2022-09-18 23:06:49 +03:00
parser . add_argument ( " --autolaunch " , action = ' store_true ' , help = " open the webui URL in the system ' s default browser upon launch " , default = False )
2022-09-20 20:10:01 +03:00
parser . add_argument ( " --use-textbox-seed " , action = ' store_true ' , help = " use textbox for seeds in UI (no up/down, but possible to input long seeds) " , default = False )
2022-10-02 20:23:40 +03:00
parser . add_argument ( " --disable-console-progressbars " , action = ' store_true ' , help = " do not output progressbars to console " , default = False )
parser . add_argument ( " --enable-console-prompts " , action = ' store_true ' , help = " print prompts to console when generating with txt2img and img2img " , default = False )
2022-10-10 06:17:29 +03:00
parser . add_argument ( ' --vae-path ' , type = str , help = ' Path to Variational Autoencoders model ' , default = None )
2022-10-09 17:58:43 +03:00
parser . add_argument ( " --disable-safe-unpickle " , action = ' store_true ' , help = " disable checking pytorch models for malicious code " , default = False )
2022-10-02 20:23:40 +03:00
2022-09-20 20:10:01 +03:00
2022-09-06 10:58:58 +03:00
cmd_opts = parser . parse_args ( )
2022-10-04 11:24:35 +03:00
2022-10-04 12:18:17 +03:00
devices . device , devices . device_gfpgan , devices . device_bsrgan , devices . device_esrgan , devices . device_scunet , devices . device_codeformer = \
( devices . cpu if x in cmd_opts . use_cpu else devices . get_optimal_device ( ) for x in [ ' SD ' , ' GFPGAN ' , ' BSRGAN ' , ' ESRGAN ' , ' SCUNet ' , ' CodeFormer ' ] )
2022-10-04 11:24:35 +03:00
device = devices . device
2022-09-11 08:11:27 +03:00
2022-09-03 12:08:45 +03:00
batch_cond_uncond = cmd_opts . always_batch_cond_uncond or not ( cmd_opts . lowvram or cmd_opts . medvram )
2022-09-06 23:10:12 +03:00
parallel_processing_allowed = not cmd_opts . lowvram and not cmd_opts . medvram
2022-10-08 16:20:41 +03:00
xformers_available = False
2022-09-12 09:44:45 +03:00
config_filename = cmd_opts . ui_settings_file
2022-09-07 12:32:28 +03:00
2022-10-12 07:01:20 +03:00
os . makedirs ( cmd_opts . hypernetwork_dir , exist_ok = True )
2022-10-11 14:53:02 +03:00
hypernetworks = hypernetwork . list_hypernetworks ( cmd_opts . hypernetwork_dir )
2022-10-09 05:37:35 +03:00
loaded_hypernetwork = None
2022-10-07 10:17:52 +03:00
2022-09-27 19:01:13 +03:00
2022-10-11 14:53:02 +03:00
def reload_hypernetworks ( ) :
global hypernetworks
hypernetworks = hypernetwork . list_hypernetworks ( cmd_opts . hypernetwork_dir )
hypernetwork . load_hypernetwork ( opts . sd_hypernetwork )
2022-09-03 12:08:45 +03:00
class State :
2022-10-05 06:56:30 +03:00
skipped = False
2022-09-03 12:08:45 +03:00
interrupted = False
job = " "
2022-09-06 02:09:01 +03:00
job_no = 0
job_count = 0
2022-09-27 04:26:13 +03:00
job_timestamp = ' 0 '
2022-09-06 02:09:01 +03:00
sampling_step = 0
sampling_steps = 0
2022-09-06 19:33:51 +03:00
current_latent = None
current_image = None
2022-09-06 23:10:12 +03:00
current_image_sampling_step = 0
2022-10-02 15:03:39 +03:00
textinfo = None
2022-09-06 23:10:12 +03:00
2022-10-05 06:56:30 +03:00
def skip ( self ) :
self . skipped = True
2022-09-03 12:08:45 +03:00
def interrupt ( self ) :
self . interrupted = True
2022-09-06 02:09:01 +03:00
def nextjob ( self ) :
self . job_no + = 1
self . sampling_step = 0
2022-09-06 23:10:12 +03:00
self . current_image_sampling_step = 0
2022-09-27 04:26:13 +03:00
2022-09-25 15:45:20 +03:00
def get_job_timestamp ( self ) :
2022-10-02 15:03:39 +03:00
return datetime . datetime . now ( ) . strftime ( " % Y % m %d % H % M % S " ) # shouldn't this return job_timestamp?
2022-09-06 02:09:01 +03:00
2022-09-06 10:11:25 +03:00
2022-09-03 12:08:45 +03:00
state = State ( )
2022-09-05 23:08:06 +03:00
artist_db = modules . artists . ArtistsDatabase ( os . path . join ( script_path , ' artists.csv ' ) )
2022-09-14 01:18:07 +03:00
styles_filename = cmd_opts . styles_file
2022-09-14 17:56:21 +03:00
prompt_styles = modules . styles . StyleDatabase ( styles_filename )
2022-09-03 12:08:45 +03:00
2022-09-11 18:48:36 +03:00
interrogator = modules . interrogate . InterrogateModels ( " interrogate " )
2022-09-09 23:16:02 +03:00
face_restorers = [ ]
2022-10-08 13:38:57 +03:00
2022-09-17 12:05:04 +03:00
2022-09-22 07:59:27 +03:00
def realesrgan_models_names ( ) :
import modules . realesrgan_model
2022-09-30 01:46:23 +03:00
return [ x . name for x in modules . realesrgan_model . get_realesrgan_models ( None ) ]
2022-09-22 07:59:27 +03:00
2022-09-22 20:41:22 +03:00
class OptionInfo :
2022-10-13 19:22:41 +03:00
def __init__ ( self , default = None , label = " " , component = None , component_args = None , onchange = None , show_on_main_page = False , refresh = None ) :
2022-09-22 20:41:22 +03:00
self . default = default
self . label = label
self . component = component
self . component_args = component_args
self . onchange = onchange
2022-09-22 21:32:44 +03:00
self . section = None
2022-10-13 19:22:41 +03:00
self . refresh = refresh
2022-09-22 20:41:22 +03:00
2022-10-08 22:12:24 +03:00
def options_section ( section_identifier , options_dict ) :
2022-09-22 20:41:22 +03:00
for k , v in options_dict . items ( ) :
2022-10-08 22:12:24 +03:00
v . section = section_identifier
2022-09-22 20:41:22 +03:00
return options_dict
2022-09-25 13:56:32 +03:00
2022-09-24 16:29:20 +03:00
hide_dirs = { " visible " : not cmd_opts . hide_ui_dir_config }
2022-09-22 20:41:22 +03:00
options_templates = { }
2022-09-22 21:32:44 +03:00
options_templates . update ( options_section ( ( ' saving-images ' , " Saving images/grids " ) , {
" samples_save " : OptionInfo ( True , " Always save all generated images " ) ,
" samples_format " : OptionInfo ( ' png ' , ' File format for images ' ) ,
" samples_filename_pattern " : OptionInfo ( " " , " Images filename pattern " ) ,
" grid_save " : OptionInfo ( True , " Always save all generated image grids " ) ,
" grid_format " : OptionInfo ( ' png ' , ' File format for grids ' ) ,
" grid_extended_filename " : OptionInfo ( False , " Add extended info (seed, prompt) to filename when saving grid " ) ,
" grid_only_if_multiple " : OptionInfo ( True , " Do not save grids consisting of one picture " ) ,
" n_rows " : OptionInfo ( - 1 , " Grid row count; use -1 for autodetect and 0 for it to be same as batch size " , gr . Slider , { " minimum " : - 1 , " maximum " : 16 , " step " : 1 } ) ,
2022-09-22 20:41:22 +03:00
" enable_pnginfo " : OptionInfo ( True , " Save text information about generation parameters as chunks to png files " ) ,
" save_txt " : OptionInfo ( False , " Create a text file next to every image with generation parameters. " ) ,
" save_images_before_face_restoration " : OptionInfo ( False , " Save a copy of image before doing face restoration. " ) ,
2022-09-22 21:32:44 +03:00
" jpeg_quality " : OptionInfo ( 80 , " Quality for saved jpeg images " , gr . Slider , { " minimum " : 1 , " maximum " : 100 , " step " : 1 } ) ,
" export_for_4chan " : OptionInfo ( True , " If PNG image is larger than 4MB or any dimension is larger than 4000, downscale and save copy as JPG " ) ,
" use_original_name_batch " : OptionInfo ( False , " Use original name for output filename during batch process in extras tab " ) ,
2022-09-28 17:05:23 +03:00
" save_selected_only " : OptionInfo ( True , " When using ' Save ' button, only save a single selected image " ) ,
2022-10-10 23:15:48 +03:00
" do_not_add_watermark " : OptionInfo ( False , " Do not add watermark to images " ) ,
2022-09-22 20:41:22 +03:00
} ) )
2022-09-22 21:32:44 +03:00
options_templates . update ( options_section ( ( ' saving-paths ' , " Paths for saving " ) , {
" outdir_samples " : OptionInfo ( " " , " Output directory for images; if empty, defaults to three directories below " , component_args = hide_dirs ) ,
2022-09-22 20:41:22 +03:00
" outdir_txt2img_samples " : OptionInfo ( " outputs/txt2img-images " , ' Output directory for txt2img images ' , component_args = hide_dirs ) ,
" outdir_img2img_samples " : OptionInfo ( " outputs/img2img-images " , ' Output directory for img2img images ' , component_args = hide_dirs ) ,
" outdir_extras_samples " : OptionInfo ( " outputs/extras-images " , ' Output directory for images from extras tab ' , component_args = hide_dirs ) ,
" outdir_grids " : OptionInfo ( " " , " Output directory for grids; if empty, defaults to two directories below " , component_args = hide_dirs ) ,
" outdir_txt2img_grids " : OptionInfo ( " outputs/txt2img-grids " , ' Output directory for txt2img grids ' , component_args = hide_dirs ) ,
" outdir_img2img_grids " : OptionInfo ( " outputs/img2img-grids " , ' Output directory for img2img grids ' , component_args = hide_dirs ) ,
" outdir_save " : OptionInfo ( " log/images " , " Directory for saving images using the Save button " , component_args = hide_dirs ) ,
} ) )
2022-09-22 21:32:44 +03:00
options_templates . update ( options_section ( ( ' saving-to-dirs ' , " Saving to a directory " ) , {
" save_to_dirs " : OptionInfo ( False , " Save images to a subdirectory " ) ,
2022-10-02 21:50:14 +03:00
" grid_save_to_dirs " : OptionInfo ( False , " Save grids to a subdirectory " ) ,
" use_save_to_dirs_for_ui " : OptionInfo ( False , " When using \" Save \" button, save images to a subdirectory " ) ,
2022-09-22 21:32:44 +03:00
" directories_filename_pattern " : OptionInfo ( " " , " Directory name pattern " ) ,
2022-10-02 21:50:14 +03:00
" directories_max_prompt_words " : OptionInfo ( 8 , " Max prompt words for [prompt_words] pattern " , gr . Slider , { " minimum " : 1 , " maximum " : 20 , " step " : 1 } ) ,
2022-09-22 21:32:44 +03:00
} ) )
options_templates . update ( options_section ( ( ' upscaling ' , " Upscaling " ) , {
" ESRGAN_tile " : OptionInfo ( 192 , " Tile size for ESRGAN upscalers. 0 = no tiling. " , gr . Slider , { " minimum " : 0 , " maximum " : 512 , " step " : 16 } ) ,
" ESRGAN_tile_overlap " : OptionInfo ( 8 , " Tile overlap, in pixels for ESRGAN upscalers. Low values = visible seam. " , gr . Slider , { " minimum " : 0 , " maximum " : 48 , " step " : 1 } ) ,
2022-09-30 01:46:23 +03:00
" realesrgan_enabled_models " : OptionInfo ( [ " R-ESRGAN x4+ " , " R-ESRGAN x4+ Anime6B " ] , " Select which Real-ESRGAN models to show in the web UI. (Requires restart) " , gr . CheckboxGroup , lambda : { " choices " : realesrgan_models_names ( ) } ) ,
2022-09-22 21:32:44 +03:00
" SWIN_tile " : OptionInfo ( 192 , " Tile size for all SwinIR. " , gr . Slider , { " minimum " : 16 , " maximum " : 512 , " step " : 16 } ) ,
" SWIN_tile_overlap " : OptionInfo ( 8 , " Tile overlap, in pixels for SwinIR. Low values = visible seam. " , gr . Slider , { " minimum " : 0 , " maximum " : 48 , " step " : 1 } ) ,
" ldsr_steps " : OptionInfo ( 100 , " LDSR processing steps. Lower = faster " , gr . Slider , { " minimum " : 1 , " maximum " : 200 , " step " : 1 } ) ,
2022-10-03 07:57:59 +03:00
" upscaler_for_img2img " : OptionInfo ( None , " Upscaler for img2img " , gr . Dropdown , lambda : { " choices " : [ x . name for x in sd_upscalers ] } ) ,
2022-09-22 21:32:44 +03:00
} ) )
options_templates . update ( options_section ( ( ' face-restoration ' , " Face restoration " ) , {
" face_restoration_model " : OptionInfo ( None , " Face restoration model " , gr . Radio , lambda : { " choices " : [ x . name ( ) for x in face_restorers ] } ) ,
" code_former_weight " : OptionInfo ( 0.5 , " CodeFormer weight parameter; 0 = maximum effect; 1 = minimum effect " , gr . Slider , { " minimum " : 0 , " maximum " : 1 , " step " : 0.01 } ) ,
" face_restoration_unload " : OptionInfo ( False , " Move face restoration model from VRAM into RAM after processing " ) ,
2022-09-22 20:41:22 +03:00
} ) )
2022-09-22 21:32:44 +03:00
options_templates . update ( options_section ( ( ' system ' , " System " ) , {
" memmon_poll_rate " : OptionInfo ( 8 , " VRAM usage polls per second during generation. Set to 0 to disable. " , gr . Slider , { " minimum " : 0 , " maximum " : 40 , " step " : 1 } ) ,
" samples_log_stdout " : OptionInfo ( False , " Always print all generation info to standard output " ) ,
2022-10-04 14:38:45 +03:00
" multiple_tqdm " : OptionInfo ( True , " Add a second progress bar to the console that shows progress for an entire job. " ) ,
2022-09-22 20:41:22 +03:00
} ) )
2022-10-11 19:03:08 +03:00
options_templates . update ( options_section ( ( ' training ' , " Training " ) , {
2022-10-12 03:21:52 +03:00
" unload_models_when_training " : OptionInfo ( False , " Unload VAE and CLIP from VRAM when training " ) ,
2022-10-12 20:49:47 +03:00
" dataset_filename_word_regex " : OptionInfo ( " " , " Filename word regex " ) ,
" dataset_filename_join_string " : OptionInfo ( " " , " Filename join string " ) ,
" training_image_repeats_per_epoch " : OptionInfo ( 100 , " Number of repeats for a single input image per epoch; used only for displaying epoch number " , gr . Number , { " precision " : 0 } ) ,
2022-10-11 19:03:08 +03:00
} ) )
2022-09-22 21:32:44 +03:00
options_templates . update ( options_section ( ( ' sd ' , " Stable Diffusion " ) , {
2022-10-13 19:22:41 +03:00
" sd_model_checkpoint " : OptionInfo ( None , " Stable Diffusion checkpoint " , gr . Dropdown , lambda : { " choices " : modules . sd_models . checkpoint_tiles ( ) } , refresh = sd_models . list_models ) ,
" sd_hypernetwork " : OptionInfo ( " None " , " Stable Diffusion finetune hypernetwork " , gr . Dropdown , lambda : { " choices " : [ " None " ] + [ x for x in hypernetworks . keys ( ) ] } , refresh = reload_hypernetworks ) ,
2022-09-22 20:41:22 +03:00
" img2img_color_correction " : OptionInfo ( False , " Apply color correction to img2img results to match original colors. " ) ,
2022-09-27 19:01:13 +03:00
" save_images_before_color_correction " : OptionInfo ( False , " Save a copy of image before applying color correction to img2img results " ) ,
2022-09-22 20:41:22 +03:00
" img2img_fix_steps " : OptionInfo ( False , " With img2img, do exactly the amount of steps the slider specifies (normally you ' d do less with less denoising). " ) ,
" enable_quantization " : OptionInfo ( False , " Enable quantization in K samplers for sharper and cleaner results. This may change existing seeds. Requires restart to apply. " ) ,
2022-10-04 14:38:45 +03:00
" enable_emphasis " : OptionInfo ( True , " Emphasis: use (text) to make model pay more attention to text and [text] to make it pay less attention " ) ,
2022-09-29 11:31:48 +03:00
" use_old_emphasis_implementation " : OptionInfo ( False , " Use old emphasis implementation. Can be useful to reproduce old seeds. " ) ,
2022-09-22 20:41:22 +03:00
" enable_batch_seeds " : OptionInfo ( True , " Make K-diffusion samplers produce same images in a batch as when making a single image " ) ,
2022-10-11 09:55:28 +03:00
" comma_padding_backtrack " : OptionInfo ( 20 , " Increase coherency by padding from the last comma within n tokens when using more than 75 tokens " , gr . Slider , { " minimum " : 0 , " maximum " : 74 , " step " : 1 } ) ,
2022-09-22 21:32:44 +03:00
" filter_nsfw " : OptionInfo ( False , " Filter NSFW content " ) ,
2022-10-08 23:33:06 +03:00
' CLIP_stop_at_last_layers ' : OptionInfo ( 1 , " Stop At last layers of CLIP model " , gr . Slider , { " minimum " : 1 , " maximum " : 12 , " step " : 1 } ) ,
2022-09-22 21:32:44 +03:00
" random_artist_categories " : OptionInfo ( [ ] , " Allowed categories for random artists selection when using the Roll button " , gr . CheckboxGroup , { " choices " : artist_db . categories ( ) } ) ,
2022-10-13 16:07:18 +03:00
' quicksettings ' : OptionInfo ( " sd_model_checkpoint " , " Quicksettings list " ) ,
2022-09-22 20:41:22 +03:00
} ) )
2022-10-12 09:00:07 +03:00
options_templates . update ( options_section ( ( ' interrogate ' , " Interrogate Options " ) , {
2022-09-22 20:41:22 +03:00
" interrogate_keep_models_in_memory " : OptionInfo ( False , " Interrogate: keep models in VRAM " ) ,
" interrogate_use_builtin_artists " : OptionInfo ( True , " Interrogate: use artists from artists.csv " ) ,
Interrogate: add option to include ranks in output
Since the UI also allows users to specify ranks, it can be useful to show people what ranks are being returned by interrogate
This can also give much better results when feeding the interrogate results back into either img2img or txt2img, especially when trying to generate a specific character or scene for which you have a similar concept image
Testing Steps:
Launch Webui with command line arg: --deepdanbooru
Navigate to img2img tab, use interrogate DeepBooru, verify tags appears as before. Use "Interrogate CLIP", verify prompt appears as before
Navigate to Settings tab, enable new option, click "apply settings"
Navigate to img2img, Interrogate DeepBooru again, verify that weights appear and are properly formatted. Note that "Interrogate CLIP" prompt is still unchanged
In my testing, this change has no effect to "Interrogate CLIP", as it seems to generate a sentence-structured caption, and not a set of tags.
(reproduce changes from https://github.com/AUTOMATIC1111/stable-diffusion-webui/pull/2149/commits/6ed4faac46c45ca7353f228aca9b436bbaba7bc7)
2022-10-12 04:02:41 +03:00
" interrogate_return_ranks " : OptionInfo ( False , " Interrogate: include ranks of model tags matches in results (Has no effect on caption-based interrogators). " ) ,
2022-09-22 20:41:22 +03:00
" interrogate_clip_num_beams " : OptionInfo ( 1 , " Interrogate: num_beams for BLIP " , gr . Slider , { " minimum " : 1 , " maximum " : 16 , " step " : 1 } ) ,
" interrogate_clip_min_length " : OptionInfo ( 24 , " Interrogate: minimum description length (excluding artists, etc..) " , gr . Slider , { " minimum " : 1 , " maximum " : 128 , " step " : 1 } ) ,
" interrogate_clip_max_length " : OptionInfo ( 48 , " Interrogate: maximum description length " , gr . Slider , { " minimum " : 1 , " maximum " : 256 , " step " : 1 } ) ,
2022-10-13 07:40:03 +03:00
" interrogate_clip_dict_limit " : OptionInfo ( 1500 , " CLIP: maximum number of lines in text file (0 = No limit) " ) ,
2022-10-12 09:00:07 +03:00
" interrogate_deepbooru_score_threshold " : OptionInfo ( 0.5 , " Interrogate: deepbooru score threshold " , gr . Slider , { " minimum " : 0 , " maximum " : 1 , " step " : 0.01 } ) ,
" deepbooru_sort_alpha " : OptionInfo ( True , " Interrogate: deepbooru sort alphabetically " ) ,
2022-10-12 21:55:43 +03:00
" deepbooru_use_spaces " : OptionInfo ( False , " use spaces for tags in deepbooru " ) ,
" deepbooru_escape " : OptionInfo ( True , " escape ( \\ ) brackets in deepbooru (so they are used as literal brackets and not for emphasis) " ) ,
2022-10-12 09:00:07 +03:00
} ) )
2022-09-22 20:41:22 +03:00
2022-09-22 21:32:44 +03:00
options_templates . update ( options_section ( ( ' ui ' , " User interface " ) , {
" show_progressbar " : OptionInfo ( True , " Show progressbar " ) ,
2022-10-09 01:18:45 +03:00
" show_progress_every_n_steps " : OptionInfo ( 0 , " Show image creation progress every N sampling steps. Set 0 to disable. " , gr . Slider , { " minimum " : 0 , " maximum " : 32 , " step " : 1 } ) ,
2022-09-22 21:32:44 +03:00
" return_grid " : OptionInfo ( True , " Show grid in results for web " ) ,
2022-10-04 17:23:48 +03:00
" do_not_show_images " : OptionInfo ( False , " Do not show any images in results for web " ) ,
2022-09-22 21:32:44 +03:00
" add_model_hash_to_info " : OptionInfo ( True , " Add model hash to generation information " ) ,
2022-10-09 14:57:48 +03:00
" add_model_name_to_info " : OptionInfo ( False , " Add model name to generation information " ) ,
2022-09-22 21:32:44 +03:00
" font " : OptionInfo ( " " , " Font for image grids that have text " ) ,
" js_modal_lightbox " : OptionInfo ( True , " Enable full page image viewer " ) ,
2022-10-08 22:12:24 +03:00
" js_modal_lightbox_initially_zoomed " : OptionInfo ( True , " Show images zoomed in by default in full page image viewer " ) ,
2022-10-06 03:35:07 +03:00
" show_progress_in_title " : OptionInfo ( True , " Show generation progress in window title. " ) ,
2022-09-22 21:32:44 +03:00
} ) )
2022-09-26 10:00:11 +03:00
options_templates . update ( options_section ( ( ' sampler-params ' , " Sampler parameters " ) , {
2022-10-06 12:08:48 +03:00
" hide_samplers " : OptionInfo ( [ ] , " Hide samplers in user interface (requires restart) " , gr . CheckboxGroup , lambda : { " choices " : [ x . name for x in sd_samplers . all_samplers ] } ) ,
" eta_ddim " : OptionInfo ( 0.0 , " eta (noise multiplier) for DDIM " , gr . Slider , { " minimum " : 0.0 , " maximum " : 1.0 , " step " : 0.01 } ) ,
" eta_ancestral " : OptionInfo ( 1.0 , " eta (noise multiplier) for ancestral samplers " , gr . Slider , { " minimum " : 0.0 , " maximum " : 1.0 , " step " : 0.01 } ) ,
" ddim_discretize " : OptionInfo ( ' uniform ' , " img2img DDIM discretize " , gr . Radio , { " choices " : [ ' uniform ' , ' quad ' ] } ) ,
' s_churn ' : OptionInfo ( 0.0 , " sigma churn " , gr . Slider , { " minimum " : 0.0 , " maximum " : 1.0 , " step " : 0.01 } ) ,
' s_tmin ' : OptionInfo ( 0.0 , " sigma tmin " , gr . Slider , { " minimum " : 0.0 , " maximum " : 1.0 , " step " : 0.01 } ) ,
' s_noise ' : OptionInfo ( 1.0 , " sigma noise " , gr . Slider , { " minimum " : 0.0 , " maximum " : 1.0 , " step " : 0.01 } ) ,
2022-10-10 20:32:37 +03:00
' eta_noise_seed_delta ' : OptionInfo ( 0 , " Eta noise seed delta " , gr . Number , { " precision " : 0 } ) ,
2022-09-26 10:00:11 +03:00
} ) )
2022-09-03 12:08:45 +03:00
2022-10-06 12:08:48 +03:00
2022-09-22 20:41:22 +03:00
class Options :
2022-09-03 12:08:45 +03:00
data = None
2022-09-22 20:41:22 +03:00
data_labels = options_templates
2022-09-23 17:27:30 +03:00
typemap = { int : float }
2022-09-03 12:08:45 +03:00
def __init__ ( self ) :
self . data = { k : v . default for k , v in self . data_labels . items ( ) }
def __setattr__ ( self , key , value ) :
if self . data is not None :
if key in self . data :
self . data [ key ] = value
return super ( Options , self ) . __setattr__ ( key , value )
def __getattr__ ( self , item ) :
if self . data is not None :
if item in self . data :
return self . data [ item ]
if item in self . data_labels :
return self . data_labels [ item ] . default
return super ( Options , self ) . __getattribute__ ( item )
def save ( self , filename ) :
with open ( filename , " w " , encoding = " utf8 " ) as file :
json . dump ( self . data , file )
2022-09-23 17:27:30 +03:00
def same_type ( self , x , y ) :
if x is None or y is None :
return True
2022-09-23 09:15:00 +03:00
2022-09-23 17:27:30 +03:00
type_x = self . typemap . get ( type ( x ) , type ( x ) )
type_y = self . typemap . get ( type ( y ) , type ( y ) )
2022-09-23 09:15:00 +03:00
2022-09-23 17:27:30 +03:00
return type_x == type_y
2022-09-23 09:15:00 +03:00
2022-09-23 17:27:30 +03:00
def load ( self , filename ) :
with open ( filename , " r " , encoding = " utf8 " ) as file :
self . data = json . load ( file )
2022-09-23 09:15:00 +03:00
bad_settings = 0
for k , v in self . data . items ( ) :
info = self . data_labels . get ( k , None )
2022-09-23 17:27:30 +03:00
if info is not None and not self . same_type ( info . default , v ) :
2022-09-23 09:15:00 +03:00
print ( f " Warning: bad setting value: { k } : { v } ( { type ( v ) . __name__ } ; expected { type ( info . default ) . __name__ } ) " , file = sys . stderr )
bad_settings + = 1
if bad_settings > 0 :
print ( f " The program is likely to not work with bad settings. \n Settings file: { filename } \n Either fix the file, or delete it and restart. " , file = sys . stderr )
2022-09-17 12:05:04 +03:00
def onchange ( self , key , func ) :
item = self . data_labels . get ( key )
item . onchange = func
2022-09-18 22:25:18 +03:00
def dumpjson ( self ) :
d = { k : self . data . get ( k , self . data_labels . get ( k ) . default ) for k in self . data_labels . keys ( ) }
return json . dumps ( d )
2022-09-03 12:08:45 +03:00
opts = Options ( )
if os . path . exists ( config_filename ) :
opts . load ( config_filename )
2022-09-04 18:54:12 +03:00
sd_upscalers = [ ]
2022-09-03 12:08:45 +03:00
sd_model = None
2022-09-05 23:08:06 +03:00
2022-09-08 16:37:13 +03:00
progress_print_out = sys . stdout
2022-09-05 23:08:06 +03:00
2022-09-08 16:37:13 +03:00
class TotalTQDM :
def __init__ ( self ) :
self . _tqdm = None
def reset ( self ) :
self . _tqdm = tqdm . tqdm (
desc = " Total progress " ,
total = state . job_count * state . sampling_steps ,
position = 1 ,
file = progress_print_out
)
def update ( self ) :
2022-10-02 20:23:40 +03:00
if not opts . multiple_tqdm or cmd_opts . disable_console_progressbars :
2022-09-08 16:37:13 +03:00
return
if self . _tqdm is None :
self . reset ( )
self . _tqdm . update ( )
2022-09-24 08:09:59 +03:00
def updateTotal ( self , new_total ) :
2022-10-02 20:23:40 +03:00
if not opts . multiple_tqdm or cmd_opts . disable_console_progressbars :
2022-09-24 08:09:59 +03:00
return
if self . _tqdm is None :
self . reset ( )
self . _tqdm . total = new_total
2022-09-08 16:37:13 +03:00
def clear ( self ) :
if self . _tqdm is not None :
self . _tqdm . close ( )
self . _tqdm = None
total_tqdm = TotalTQDM ( )
2022-09-17 07:49:31 +03:00
mem_mon = modules . memmon . MemUsageMonitor ( " MemMon " , device , opts )
mem_mon . start ( )