2022-09-03 03:08:45 -06:00
import argparse
2022-09-29 16:46:23 -06:00
import datetime
2022-09-03 03:08:45 -06:00
import json
import os
2022-09-29 16:46:23 -06:00
import sys
2022-09-03 03:08:45 -06:00
import gradio as gr
2022-09-08 07:37:13 -06:00
import tqdm
2022-09-03 03:08:45 -06:00
2022-09-05 14:08:06 -06:00
import modules . artists
2022-09-11 09:48:36 -06:00
import modules . interrogate
2022-09-16 22:49:31 -06:00
import modules . memmon
2022-09-17 03:05:04 -06:00
import modules . sd_models
2022-09-29 16:46:23 -06:00
import modules . styles
2022-10-04 02:24:35 -06:00
import modules . devices as devices
2022-10-06 03:08:48 -06:00
from modules import sd_samplers
2022-09-29 16:46:23 -06:00
from modules . paths import script_path , sd_path
2022-09-03 03:08:45 -06:00
sd_model_file = os . path . join ( script_path , ' model.ckpt ' )
2022-09-17 03:05:04 -06:00
default_sd_model_file = sd_model_file
2022-09-26 08:29:50 -06:00
model_path = os . path . join ( script_path , ' models ' )
2022-09-03 03:08:45 -06: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 13:16:03 -06: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 02:42:40 -06:00
parser . add_argument ( " --ckpt-dir " , type = str , default = None , help = " Path to directory with stable diffusion checkpoints " )
2022-09-03 03:08:45 -06: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 01:26:00 -06:00
parser . add_argument ( " --gfpgan-model " , type = str , help = " GFPGAN model file name " , default = None )
2022-09-03 03:08:45 -06:00
parser . add_argument ( " --no-half " , action = ' store_true ' , help = " do not switch the model to 16-bit floats " )
2022-09-14 21:04:05 -06: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 03:08:45 -06:00
parser . add_argument ( " --max-batch-count " , type = int , default = 16 , help = " maximum batch count value for the UI " )
2022-09-13 00:42:51 -06:00
parser . add_argument ( " --embeddings-dir " , type = str , default = os . path . join ( script_path , ' embeddings ' ) , help = " embeddings directory for textual inversion (default: embeddings) " )
2022-09-03 03:08:45 -06:00
parser . add_argument ( " --allow-code " , action = ' store_true ' , help = " allow custom script execution from webui " )
2022-09-08 17:17:38 -06: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 02:14:42 -06: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 09:40:06 -06:00
parser . add_argument ( " --unload-gfpgan " , action = ' store_true ' , help = " does not do anything. " )
2022-09-03 03:08:45 -06: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-09-27 10:01:13 -06:00
parser . add_argument ( " --codeformer-models-path " , type = str , help = " Path to directory with codeformer model file(s). " , default = os . path . join ( model_path , ' Codeformer ' ) )
parser . add_argument ( " --gfpgan-models-path " , type = str , help = " Path to directory with GFPGAN model file(s). " , default = os . path . join ( model_path , ' GFPGAN ' ) )
parser . add_argument ( " --esrgan-models-path " , type = str , help = " Path to directory with ESRGAN model file(s). " , default = os . path . join ( model_path , ' ESRGAN ' ) )
2022-09-29 16:46:23 -06:00
parser . add_argument ( " --bsrgan-models-path " , type = str , help = " Path to directory with BSRGAN model file(s). " , default = os . path . join ( model_path , ' BSRGAN ' ) )
2022-09-27 10:01:13 -06:00
parser . add_argument ( " --realesrgan-models-path " , type = str , help = " Path to directory with RealESRGAN model file(s). " , default = os . path . join ( model_path , ' RealESRGAN ' ) )
2022-09-30 14:28:37 -06:00
parser . add_argument ( " --scunet-models-path " , type = str , help = " Path to directory with ScuNET model file(s). " , default = os . path . join ( model_path , ' ScuNET ' ) )
2022-09-27 10:01:13 -06:00
parser . add_argument ( " --swinir-models-path " , type = str , help = " Path to directory with SwinIR model file(s). " , default = os . path . join ( model_path , ' SwinIR ' ) )
parser . add_argument ( " --ldsr-models-path " , type = str , help = " Path to directory with LDSR model file(s). " , default = os . path . join ( model_path , ' LDSR ' ) )
2022-09-21 00:49:02 -06:00
parser . add_argument ( " --opt-split-attention " , action = ' store_true ' , help = " force-enables cross-attention layer optimization. By default, it ' s on for torch.cuda and off for other torch devices. " )
parser . add_argument ( " --disable-opt-split-attention " , action = ' store_true ' , help = " force-disables cross-attention layer optimization " )
2022-09-18 15:39:41 -06: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-04 03:22:50 -06: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 10:37:11 -06: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 02:42:21 -06: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 06:41:29 -06:00
parser . add_argument ( " --show-negative-prompt " , action = ' store_true ' , help = " does not do anything " , default = False )
2022-09-09 23:18:54 -06: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 14:00:42 -06:00
parser . add_argument ( " --hide-ui-dir-config " , action = ' store_true ' , help = " hide directory configuration from webui " , default = False )
2022-09-12 00:44:45 -06: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 03:40:55 -06:00
parser . add_argument ( " --gradio-debug " , action = ' store_true ' , help = " launch gradio with --debug option " )
2022-09-12 06:19:14 -06: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-05 20:11:15 -06: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 07:43:11 -06:00
parser . add_argument ( " --opt-channelslast " , action = ' store_true ' , help = " change memory type for stable diffusion to channels last " )
2022-09-13 16:18:07 -06: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 14:06:49 -06: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 11:10:01 -06: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 11:23:40 -06: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-09-20 11:10:01 -06:00
2022-09-06 01:58:58 -06:00
cmd_opts = parser . parse_args ( )
2022-10-04 02:24:35 -06:00
2022-10-04 03:18:17 -06: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 02:24:35 -06:00
device = devices . device
2022-09-10 23:11:27 -06:00
2022-09-03 03:08:45 -06:00
batch_cond_uncond = cmd_opts . always_batch_cond_uncond or not ( cmd_opts . lowvram or cmd_opts . medvram )
2022-09-06 14:10:12 -06:00
parallel_processing_allowed = not cmd_opts . lowvram and not cmd_opts . medvram
2022-09-06 10:33:51 -06:00
2022-09-12 00:44:45 -06:00
config_filename = cmd_opts . ui_settings_file
2022-09-07 03:32:28 -06:00
2022-09-27 10:01:13 -06:00
2022-09-03 03:08:45 -06:00
class State :
interrupted = False
job = " "
2022-09-05 17:09:01 -06:00
job_no = 0
job_count = 0
2022-09-26 19:26:13 -06:00
job_timestamp = ' 0 '
2022-09-05 17:09:01 -06:00
sampling_step = 0
sampling_steps = 0
2022-09-06 10:33:51 -06:00
current_latent = None
current_image = None
2022-09-06 14:10:12 -06:00
current_image_sampling_step = 0
2022-10-02 06:03:39 -06:00
textinfo = None
2022-09-06 14:10:12 -06:00
2022-09-03 03:08:45 -06:00
def interrupt ( self ) :
self . interrupted = True
2022-09-05 17:09:01 -06:00
def nextjob ( self ) :
self . job_no + = 1
self . sampling_step = 0
2022-09-06 14:10:12 -06:00
self . current_image_sampling_step = 0
2022-09-26 19:26:13 -06:00
2022-09-25 06:45:20 -06:00
def get_job_timestamp ( self ) :
2022-10-02 06:03:39 -06:00
return datetime . datetime . now ( ) . strftime ( " % Y % m %d % H % M % S " ) # shouldn't this return job_timestamp?
2022-09-05 17:09:01 -06:00
2022-09-06 01:11:25 -06:00
2022-09-03 03:08:45 -06:00
state = State ( )
2022-09-05 14:08:06 -06:00
artist_db = modules . artists . ArtistsDatabase ( os . path . join ( script_path , ' artists.csv ' ) )
2022-09-13 16:18:07 -06:00
styles_filename = cmd_opts . styles_file
2022-09-14 08:56:21 -06:00
prompt_styles = modules . styles . StyleDatabase ( styles_filename )
2022-09-03 03:08:45 -06:00
2022-09-11 09:48:36 -06:00
interrogator = modules . interrogate . InterrogateModels ( " interrogate " )
2022-09-09 14:16:02 -06:00
face_restorers = [ ]
2022-09-27 10:01:13 -06:00
# This was moved to webui.py with the other model "setup" calls.
# modules.sd_models.list_models()
2022-09-17 03:05:04 -06:00
2022-09-21 22:59:27 -06:00
def realesrgan_models_names ( ) :
import modules . realesrgan_model
2022-09-29 16:46:23 -06:00
return [ x . name for x in modules . realesrgan_model . get_realesrgan_models ( None ) ]
2022-09-21 22:59:27 -06:00
2022-09-22 11:41:22 -06:00
class OptionInfo :
def __init__ ( self , default = None , label = " " , component = None , component_args = None , onchange = None ) :
self . default = default
self . label = label
self . component = component
self . component_args = component_args
self . onchange = onchange
2022-09-22 12:32:44 -06:00
self . section = None
2022-09-22 11:41:22 -06:00
def options_section ( section_identifer , options_dict ) :
for k , v in options_dict . items ( ) :
v . section = section_identifer
return options_dict
2022-09-25 04:56:32 -06:00
2022-09-24 07:29:20 -06:00
hide_dirs = { " visible " : not cmd_opts . hide_ui_dir_config }
2022-09-22 11:41:22 -06:00
options_templates = { }
2022-09-22 12:32:44 -06: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 11:41:22 -06: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 12:32:44 -06: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 08:05:23 -06:00
" save_selected_only " : OptionInfo ( True , " When using ' Save ' button, only save a single selected image " ) ,
2022-09-22 11:41:22 -06:00
} ) )
2022-09-22 12:32:44 -06: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 11:41:22 -06: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 12:32:44 -06: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 12:50:14 -06: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 12:32:44 -06:00
" directories_filename_pattern " : OptionInfo ( " " , " Directory name pattern " ) ,
2022-10-02 12:50:14 -06: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 12:32:44 -06: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-29 16:46:23 -06: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 12:32:44 -06: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-02 22:57:59 -06:00
" upscaler_for_img2img " : OptionInfo ( None , " Upscaler for img2img " , gr . Dropdown , lambda : { " choices " : [ x . name for x in sd_upscalers ] } ) ,
2022-09-22 12:32:44 -06: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 11:41:22 -06:00
} ) )
2022-09-22 12:32:44 -06: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 05:38:45 -06:00
" multiple_tqdm " : OptionInfo ( True , " Add a second progress bar to the console that shows progress for an entire job. " ) ,
2022-09-22 11:41:22 -06:00
} ) )
2022-09-22 12:32:44 -06:00
options_templates . update ( options_section ( ( ' sd ' , " Stable Diffusion " ) , {
2022-09-30 03:56:36 -06:00
" sd_model_checkpoint " : OptionInfo ( None , " Stable Diffusion checkpoint " , gr . Dropdown , lambda : { " choices " : modules . sd_models . checkpoint_tiles ( ) } ) ,
2022-09-22 11:41:22 -06:00
" img2img_color_correction " : OptionInfo ( False , " Apply color correction to img2img results to match original colors. " ) ,
2022-09-27 10:01:13 -06:00
" save_images_before_color_correction " : OptionInfo ( False , " Save a copy of image before applying color correction to img2img results " ) ,
2022-09-22 11:41:22 -06: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 05:38:45 -06: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 02:31:48 -06:00
" use_old_emphasis_implementation " : OptionInfo ( False , " Use old emphasis implementation. Can be useful to reproduce old seeds. " ) ,
2022-09-22 11:41:22 -06:00
" enable_batch_seeds " : OptionInfo ( True , " Make K-diffusion samplers produce same images in a batch as when making a single image " ) ,
2022-09-22 12:32:44 -06:00
" filter_nsfw " : OptionInfo ( False , " Filter NSFW content " ) ,
" random_artist_categories " : OptionInfo ( [ ] , " Allowed categories for random artists selection when using the Roll button " , gr . CheckboxGroup , { " choices " : artist_db . categories ( ) } ) ,
2022-09-22 11:41:22 -06:00
} ) )
2022-09-22 12:32:44 -06:00
options_templates . update ( options_section ( ( ' interrogate ' , " Interrogate Options " ) , {
2022-09-22 11:41:22 -06: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_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 } ) ,
" interrogate_clip_dict_limit " : OptionInfo ( 1500 , " Interrogate: maximum number of lines in text file (0 = No limit) " ) ,
} ) )
2022-09-22 12:32:44 -06:00
options_templates . update ( options_section ( ( ' ui ' , " User interface " ) , {
" show_progressbar " : OptionInfo ( True , " Show progressbar " ) ,
" show_progress_every_n_steps " : OptionInfo ( 0 , " Show show image creation progress every N sampling steps. Set 0 to disable. " , gr . Slider , { " minimum " : 0 , " maximum " : 32 , " step " : 1 } ) ,
" return_grid " : OptionInfo ( True , " Show grid in results for web " ) ,
2022-10-04 08:23:48 -06:00
" do_not_show_images " : OptionInfo ( False , " Do not show any images in results for web " ) ,
2022-09-22 12:32:44 -06:00
" add_model_hash_to_info " : OptionInfo ( True , " Add model hash to generation information " ) ,
" font " : OptionInfo ( " " , " Font for image grids that have text " ) ,
" js_modal_lightbox " : OptionInfo ( True , " Enable full page image viewer " ) ,
" js_modal_lightbox_initialy_zoomed " : OptionInfo ( True , " Show images zoomed in by default in full page image viewer " ) ,
2022-10-05 18:09:44 -06:00
" show_progress_in_title " : OptionInfo ( False , " Show generation progress in window title. " ) ,
2022-09-22 12:32:44 -06:00
} ) )
2022-09-26 01:00:11 -06:00
options_templates . update ( options_section ( ( ' sampler-params ' , " Sampler parameters " ) , {
2022-10-06 03:08:48 -06: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-09-26 01:00:11 -06:00
} ) )
2022-09-03 03:08:45 -06:00
2022-10-06 03:08:48 -06:00
2022-09-22 11:41:22 -06:00
class Options :
2022-09-03 03:08:45 -06:00
data = None
2022-09-22 11:41:22 -06:00
data_labels = options_templates
2022-09-23 08:27:30 -06:00
typemap = { int : float }
2022-09-03 03:08:45 -06: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 08:27:30 -06:00
def same_type ( self , x , y ) :
if x is None or y is None :
return True
2022-09-23 00:15:00 -06:00
2022-09-23 08:27:30 -06:00
type_x = self . typemap . get ( type ( x ) , type ( x ) )
type_y = self . typemap . get ( type ( y ) , type ( y ) )
2022-09-23 00:15:00 -06:00
2022-09-23 08:27:30 -06:00
return type_x == type_y
2022-09-23 00:15:00 -06:00
2022-09-23 08:27:30 -06:00
def load ( self , filename ) :
with open ( filename , " r " , encoding = " utf8 " ) as file :
self . data = json . load ( file )
2022-09-23 00:15:00 -06:00
bad_settings = 0
for k , v in self . data . items ( ) :
info = self . data_labels . get ( k , None )
2022-09-23 08:27:30 -06:00
if info is not None and not self . same_type ( info . default , v ) :
2022-09-23 00:15:00 -06: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 03:05:04 -06:00
def onchange ( self , key , func ) :
item = self . data_labels . get ( key )
item . onchange = func
2022-09-18 13:25:18 -06: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 03:08:45 -06:00
opts = Options ( )
if os . path . exists ( config_filename ) :
opts . load ( config_filename )
2022-09-04 09:54:12 -06:00
sd_upscalers = [ ]
2022-09-03 03:08:45 -06:00
sd_model = None
2022-09-05 14:08:06 -06:00
2022-09-08 07:37:13 -06:00
progress_print_out = sys . stdout
2022-09-05 14:08:06 -06:00
2022-09-08 07:37:13 -06: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 11:23:40 -06:00
if not opts . multiple_tqdm or cmd_opts . disable_console_progressbars :
2022-09-08 07:37:13 -06:00
return
if self . _tqdm is None :
self . reset ( )
self . _tqdm . update ( )
2022-09-23 23:09:59 -06:00
def updateTotal ( self , new_total ) :
2022-10-02 11:23:40 -06:00
if not opts . multiple_tqdm or cmd_opts . disable_console_progressbars :
2022-09-23 23:09:59 -06:00
return
if self . _tqdm is None :
self . reset ( )
self . _tqdm . total = new_total
2022-09-08 07:37:13 -06:00
def clear ( self ) :
if self . _tqdm is not None :
self . _tqdm . close ( )
self . _tqdm = None
total_tqdm = TotalTQDM ( )
2022-09-16 22:49:31 -06:00
mem_mon = modules . memmon . MemUsageMonitor ( " MemMon " , device , opts )
mem_mon . start ( )