import gradio as gr import torch import modin.pandas as pd from PIL import Image from diffusers import DiffusionPipeline import os import random import torchsde from math import floor, copysign import time ########## # # Spaghetti AI by MagicFixesEverything # https://huggingface.co/magicfixeseverything # # This app is based on scripts by: # https://huggingface.co/Manjushri # This app has been adapted from that person's versions. # ########## # # For instructions on how to install this app offline, see the instructions # here: # https://huggingface.co/spaces/magicfixeseverything/ai_image_creation/blob/main/Instructions.txt # # The display was tested with gradio version 4.11.0. # # To launch this script, use the following in the command prompt, taking off # the # at the start. (You will need to adjust the start of the path if you # have changed the location) # #cd C:\Diffusers && .venv\Scripts\activate.bat && py .venv\ai_image_creation\app.py # # You must have a NVIDIA graphics card in your computer with Cuda installed # to use this script. It will not work on just a CPU on Windows. # # If not using "enable_model_cpu_offload" or "enable_sequential_cpu_offload", # memory usage will remain high until command prompt is closed. (whether # image is being created or not) # ############################################################################### ############################################################################### # # # # Begin Configurations # # # ############################################################################### ############################################################################### # # Main Directory # # This is where everything goes. Your Python virtual environment should # be here. Model data will be stored here. (unless you change the next # configuration) If configured, imagery will also be automatically be # saved here. # main_dir = "C:/Spaghetti_AI" #################### # # Only Use Local Files (IMPORTANT) # # Please read the section below the configuration. The app will not work # unless this is set to "0" in order for it to download model data. # # 0 False (Model data will be downloaded) # 1 True (Model data will not be downloaded, even if missing) # only_use_local_files = 0 # # This is an important value. HuggingFace doesn't just download a model # once and never try again. If something is updated, it will download it # again. It will not delete the older version. This could eventually # allow it to use all the space on your drive as that could potentially # add 5 to 15 gigabytes of data each time a model updates. This variable # forces the script not to check for model data. However, when set to # "1", the app will not download any data for the models, including for # refining and upscaling. That would mean the app will not work. You need # to set this to "0" when you need to download data. While there are ways # to not download as much data, it is not as simple. I prefer this # method. You could check occasionally and manually delete model data, # but if you're like me, you would forget. # # To download data you can use each model in the app when this is set to # "0" and it will download the needed data. That does require you to use # each model. Then you could set this back to "1". However, if you want # to download all the data, without having to use each model, including # using refining and upscaling at some point at least once, there are # two options. # # To download all default model data, meaning the default model selected # for each base model (set in # "base_model_model_configuration_defaults_object"), including data for # the refiner and upscaler, add this to the end of the url for option 1: # # ?download_data=1 # # Like this for option 1: # http://127.0.0.1:7860/?download_data=1 # # To download all model data, for everything in the # "model_configuration_links_object", including data for the refiner and # upscaler, add this to the end of the url for option 2: # # ?download_data=2 # # Both options will download dozens of gigabytes of data, most especially # the second option, so you may not want to do that. Before you do that, # make sure you have removed from the configurations the models you do # not want. For option 1, remove the base models you do not want to use # in "base_model_array". For option 2, remove the model configurations # you do not want to use in # "base_model_object_of_model_configuration_arrays". # # To have model data download, this variable must be set to 0. You must # also set "HF_HUB_OFFLINE" to "0" in "spaghetti_ai_launcher.bat" if you # use that script. If you use that script, and either is not that, model # data will not download. # #################### # # Use Custom Hugging Face Cache Directory # # The folder where model data is stored can get huge. I choose to add it # to a place where I am more likely to notice it more often. If you use # other Hugging Face things however, and will use these models in those # other things, then you might want to consider not having this here as # it would duplicate the model data. # # If set to 1, the data would be here: # C:\Diffusers\model_data # # If set to 0, the data would be here: # %USERPROFILE%/.cache/huggingface/hub # Which would look like this, where {Username} is the username of # your Windows account: # C:\Users\{Username}\.cache\huggingface\hub # # You need to clean out the folder occasionally as this folder will get # extremely large. Eventually, it would take up all the space on your # computer. # use_custom_hugging_face_cache_dir = 1 ##### # # Name of Model Data Folder # # This is where all the model data will go. (unless you changed it in the # previous configuration) This folder will get very large. You need to # clean it out manually occasionally. # cache_directory_folder_name = "model_data" #################### # # Default Base Model # # This will automatically be SDXL Turbo if you are running this on a CPU. # default_base_model = "sdxl" #################### # # Use Safety Checker # # This can block some NSFW images for the models that allow it. # # 0 No # 1 Yes # use_safety_checker = 0 ##### # # Auto Save Imagery # # You can automatically save the image file, and a text file with the # prompt details. # auto_save_imagery = 1 ##### # # Name of Saved Images Folder # # You can change the name of this folder if you want. Imagery will be # saved in a folder called "saved_images" in the directory configured # in "main_dir". (the saved images folder will be created # automatically) A directory for each day will be created in this # folder. Imagery will then be placed in each folder. # saved_images_folder_name = "saved_images" #################### # # Auto Open Browser From Command Prompt # auto_open_browser = 0 #################### # # Allow Image Generation Cancellation # # This allows canceling the image generation. It will not stop # immediately. It will stop after completing the current step it is on. # enable_image_generation_cancellation = 1 #################### # # Include Close Command Prompt / Cancel Button # # This will likely be removed in the future. # # This doesn't work well at all. It just closes the command prompt. I # might remove this eventually. # enable_close_command_prompt_button = 0 #################### # # Use Denoising Start In Base Model When Using Refiner # # If set to "1", refining will end at the percent (expressed as decimal) # defined in the denoising start for the refiner. If the steps set are # 100, and the denoising start value is 0.75, the base model will run for # 75 steps. The refiner will then run for 25 steps. # default_use_denoising_start_in_base_model_when_using_refiner = 1 #################### # # Base Model Output To Refiner Is In Latent Space # # If set to "1", base model output is in latent space instead of PIL # image when sent to refiner. # default_base_model_output_to_refiner_is_in_latent_space = 1 #################### # # Log Generation Times # # Log generation times to saved text output. The initial time it takes to # load a model is not included in the generation time. # log_generation_times = 1 #################### # # Use Image Gallery # use_image_gallery = 1 #################### # # Show Image Creation Progress Log # # This adds the current step that image generation is on. # show_image_creation_progress_log = 1 #################### # # Show Messages In Command Prompt # # Messages will be printed in command prompt. # show_messages_in_command_prompt = 1 #################### # # Show Messages In Modal On Page # # A popup appears in the top right corner on the page. # show_messages_in_modal_on_page = 0 #################### # # Suppress Hugging Face Hub Offline Status # # By default, we add messages about the current setting of # "HF_HUB_OFFLINE". # suppress_hugging_face_hub_offline_status = 0 #################### # # Add Seed Into Pipe # # To make generation deterministic. I add the option because the online # configuration for the PhotoReal site doesn't do that and it changes # things. # default_add_seed_into_pipe = 1 #################### # # Use torch.manual_seed But Do Not Add To Pipe # # We need this to match the PhotoReal site to make it work. # default_use_torch_manual_seed_but_do_not_add_to_pipe = 0 #################### # # Save Base Image When Using Refiner # # The image will be shown on the page. If you save images automatically, # it will also be saved. # default_save_base_image_when_using_refiner = 1 #################### # # Save Refined Image When Using Upscaler # # The image will be shown on the page. If you save images automatically, # it will also be saved. # default_save_refined_image_when_using_upscaler = 1 #################### # # Max Queue Size # max_queue_size_if_cpu = 3 max_queue_size_if_torch = 20 #################### # # Allow Other Model Versions # # This allows models other than the default versions. # allow_other_model_versions = 1 #################### # # Image Preview # enable_image_preview = 1 use_image_preview = 1 image_preview_step_interval = 10 image_preview_seconds_interval = 60 load_image_preview_frequency_in_seconds = 2 #################### sdxl_link = "https://huggingface.co/spaces/Manjushri/SDXL-1.0" photoreal_link = "https://huggingface.co/spaces/Manjushri/PhotoReal-V3.8.1" sdxl_turbo_link = "https://huggingface.co/spaces/diffusers/unofficial-SDXL-Turbo-i2i-t2i" #################### # # Up Next Is Various Configuration Arrays and Objects # #################### base_model_array = [ "sdxl", "photoreal", "sdxl_turbo", "sd_1_5_runwayml" ] base_model_names_object = { "sdxl": "Stable Diffusion XL", "photoreal": "PhotoReal", "sdxl_turbo": "Stable Diffusion XL Turbo", "sd_1_5_runwayml": "Stable Diffusion 1.5" } #################### # # To match very closely older the older defaults from these two links: # https://huggingface.co/spaces/Manjushri/SDXL-1.0 # https://huggingface.co/spaces/Manjushri/PhotoReal-V3.8.1 # # The PhotoReal site doesn't explicitly add the seed to the generation, so # that option needs to be unchecked to match that imagery. # # You can use the following customizations... # ########## # # Stable Diffusion XL # # The refiner always runs for SDXL at this link. # # - Valid from November 12th to present. # Number of steps in upscaler changed from 5 to 15. # # Valid from September 5th to November 12th. # Number of steps in upscaler was 5 during this period. # There were changes on this date. # # http://127.0.0.1:7860/?model=sdxl&model_config=sdxl_1-0&scheduler=model_default&width=768&height=768&guidance=10&steps=50&prompt=&neg_prompt=&seed=&add_seed=yes&refiner=yes&denoise_start=0.95&refiner_steps=100&use_denoise_end=no&latent_space_before_refiner=yes&upscaler=no&upscaler_steps=15 # ########## # # PhotoReal 3.8.1 # # - Valid from December 29th to present. # Base model was changed from 3.7.5 to 3.8.1: "circulus/canvers-real-v3.8.1" # # For some reason, I haven't been able to match this version of the # configuration yet. That means this link doesn't work: # # http://127.0.0.1:7860/?model=photoreal&model_config=photoreal_3-8-1&scheduler=model_default&width=768&height=768&guidance=5&steps=50&prompt=&neg_prompt=&seed=&add_seed=no&use_torch_manual_seed_but_not_in_generator=yes&refiner=no&denoise_start=0.95&refiner_steps=&use_denoise_end=no&latent_space_before_refiner=no&upscaler=no&upscaler_steps= # #################### # # PhotoReal 3.7.5 # # - Valid from November 12th to December 29th. # Base model was changed from 3.6 to 3.7.5: "circulus/canvers-real-v3.7.5" # # http://127.0.0.1:7860/?model=photoreal&model_config=photoreal_3-7-5&scheduler=model_default&width=768&height=768&guidance=7&steps=50&prompt=&neg_prompt=&seed=&add_seed=no&use_torch_manual_seed_but_not_in_generator=yes&refiner=no&denoise_start=0.95&refiner_steps=&use_denoise_end=no&latent_space_before_refiner=no&upscaler=no&upscaler_steps= # #################### # # PhotoReal 3.6 # # - Valid from September 1st to November 12th. # "circulus/canvers-realistic-v3.6" was already in effect. # But there were changes on this date. # # http://127.0.0.1:7860/?model=photoreal&model_config=photoreal_3-6&scheduler=model_default&width=768&height=768&guidance=7&steps=50&prompt=&neg_prompt=&seed=&add_seed=no&use_torch_manual_seed_but_not_in_generator=yes&refiner=no&denoise_start=0.95&refiner_steps=&use_denoise_end=no&latent_space_before_refiner=no&upscaler=no&upscaler_steps= # #################### base_model_object_of_model_configuration_arrays = { "sdxl": [ "sdxl_default", "sdxl_1-0" ], "photoreal": [ "photoreal_default", "photoreal_3-8-1", "photoreal_3-8", "photoreal_3-7-5", "photoreal_3-6" ], "sdxl_turbo": [ "sdxl_turbo_default", "sdxl_turbo_initial" ], "sd_1_5_runwayml": [ "sd_1_5_runwayml_default" ] } #################### model_configuration_names_object = { "sdxl_default": "Default (currently 1.0)", "sdxl_1-0": "1.0", "photoreal_default": "Default (currently 3.6)", "photoreal_3-8-1": "3.8.1", "photoreal_3-8": "3.8", "photoreal_3-7-5": "3.7.5", "photoreal_3-6": "3.6", "sdxl_turbo_default": "Default (currently Initial Release)", "sdxl_turbo_initial": "Initial Release", "sd_1_5_runwayml_default": "Default" } model_configuration_links_object = { "sdxl_default": "stabilityai/stable-diffusion-xl-base-1.0", "sdxl_1-0": "stabilityai/stable-diffusion-xl-base-1.0", "photoreal_default": "circulus/canvers-realistic-v3.6", "photoreal_3-8-1": "circulus/canvers-real-v3.8.1", "photoreal_3-8": "circulus/canvers-real-v3.8", "photoreal_3-7-5": "circulus/canvers-real-v3.7.5", "photoreal_3-6": "circulus/canvers-realistic-v3.6", "sdxl_turbo_default": "stabilityai/sdxl-turbo", "sdxl_turbo_initial": "stabilityai/sdxl-turbo", "sd_1_5_runwayml_default": "runwayml/stable-diffusion-v1-5" } #################### base_models_not_supporting_denoising_end_for_base_model_object = { "photoreal": 1, "sd_1_5_runwayml": 1 } #################### hugging_face_refiner_partial_path = "stabilityai/stable-diffusion-xl-refiner-1.0" hugging_face_upscaler_partial_path = "stabilityai/sd-x2-latent-upscaler" #################### base_model_model_configuration_defaults_object = { "sdxl": "sdxl_default", "photoreal": "photoreal_default", "sdxl_turbo": "sdxl_turbo_default", "sd_1_5_runwayml": "sd_1_5_runwayml_default" } #################### # # Links: # # SD-XL 1.0-base Model Card # https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0 # # SD-XL 1.0-refiner Model Card # https://huggingface.co/stabilityai/stable-diffusion-xl-refiner-1.0 # # Stable Diffusion x2 latent upscaler model card # https://huggingface.co/stabilityai/sd-x2-latent-upscaler # # PhotoReal # 3.8.1: https://huggingface.co/circulus/canvers-real-v3.8.1 # 3.8: https://huggingface.co/circulus/circulus/canvers-real-v3.8 # 3.7.5: https://huggingface.co/circulus/canvers-real-v3.7.5 # 3.6: https://huggingface.co/circulus/canvers-realistic-v3.6 # # SDXL Turbo # https://huggingface.co/stabilityai/sdxl-turbo # # Stable Diffusion v1-5 (runwayml) # https://huggingface.co/runwayml/stable-diffusion-v1-5 # #################### default_scheduler = "model_default" schedulers_array = [ "model_default", "ddim", "ddpm", "dpm_solver_multistep", "dpm_solver_multistep_karras_sigmas_true", "dpm_solver_multistep_algorithm_type_sde-dpmsolver_pp", "dpm_solver_multistep_karras_sigmas_true_algorithm_type_sde-dpmsolver_pp", "dpm_solver_singlestep", "dpm_solver_singlestep_karras_sigmas_true", "kdpm2_discrete", "kdpm2_discrete_karras_sigmas_true", "kdpm2_ancestral_discrete", "kdpm2_ancestral_discrete_karras_sigmas_true", "euler_discrete", "euler_ancestral_discrete", "heun_discrete", "lms_discrete", "lms_discrete_karras_sigmas_true", "pndm", "pndm_skip_prk_steps_true", "deis_multistep", "dpm_solver_sde", "uni_pc_multistep" ] scheduler_long_names_object = { "model_default": "Model Default", "ddim": "DDIM", "ddpm": "DDPM", "dpm_solver_multistep": "DPM++ 2M (DPMSolverMultistep)", "dpm_solver_multistep_karras_sigmas_true": "DPM++ 2M Karras (DPMSolverMultistep with use_karras_sigmas=True)", "dpm_solver_multistep_algorithm_type_sde-dpmsolver_pp": "DPM++ 2M SDE (DPMSolverMultistep with algorithm_type=\"sde-dpmsolver++\")", "dpm_solver_multistep_karras_sigmas_true_algorithm_type_sde-dpmsolver_pp": "DPM++ 2M SDE Karras (DPMSolverMultistep with use_karras_sigmas=True & algorithm_type=\"sde-dpmsolver++\")", "dpm_solver_singlestep": "DPM++ SDE (DPMSolverSinglestep)", "dpm_solver_singlestep_karras_sigmas_true": "DPM++ SDE Karras (DPMSolverSinglestep with use_karras_sigmas=True)", "kdpm2_discrete": "DPM2 (KDPM2Discrete)", "kdpm2_discrete_karras_sigmas_true": "DPM2 Karras (KDPM2Discrete with use_karras_sigmas=True)", "kdpm2_ancestral_discrete": "DPM2 a (KDPM2AncestralDiscrete)", "kdpm2_ancestral_discrete_karras_sigmas_true": "DPM2 a Karras (KDPM2AncestralDiscrete with use_karras_sigmas=True)", "euler_discrete": "Euler (EulerDiscrete)", "euler_ancestral_discrete": "Euler a (EulerAncestralDiscrete)", "heun_discrete": "Heun (HeunDiscrete)", "lms_discrete": "LMS (LMSDiscrete)", "lms_discrete_karras_sigmas_true": "LMS Karras (LMSDiscrete with use_karras_sigmas=True)", "pndm": "PNDM", "pndm_skip_prk_steps_true": "PNDM (with skip_prk_steps=True) - Close to PLMS", "deis_multistep": "DEISMultistep", "dpm_solver_sde": "DPMSolverSDE", "uni_pc_multistep": "UniPCMultistep" } scheduler_short_names_object = { "ddim": "DDIM", "ddpm": "DDPM", "dpm_solver_multistep": "DPM++ 2M", "dpm_solver_multistep_karras_sigmas_true": "DPM++ 2M Karras", "dpm_solver_multistep_algorithm_type_sde-dpmsolver_pp": "DPM++ 2M SDE", "dpm_solver_multistep_karras_sigmas_true_algorithm_type_sde-dpmsolver_pp": "DPM++ 2M SDE Karras", "dpm_solver_singlestep": "DPM++ SDE", "dpm_solver_singlestep_karras_sigmas_true": "DPM++ SDE Karras", "kdpm2_discrete": "DPM2", "kdpm2_discrete_karras_sigmas_true": "DPM2 Karras", "kdpm2_ancestral_discrete": "DPM2 a", "kdpm2_ancestral_discrete_karras_sigmas_true": "DPM2 a Karras", "euler_discrete": "Euler", "euler_ancestral_discrete": "Euler a", "heun_discrete": "Heun", "lms_discrete": "LMS", "lms_discrete_karras_sigmas_true": "LMS Karras", "pndm": "PNDM", "pndm_skip_prk_steps_true": "PNDM (with skip_prk_steps=True) - Close to PLMS", "deis_multistep": "DEISMultistep", "dpm_solver_sde": "DPMSolverSDE", "uni_pc_multistep": "UniPCMultistep" } scheduler_name_to_identifier_in_app_object = { "DDIMScheduler": "ddim", "DDPMScheduler": "ddpm", "DPMSolverMultistepScheduler": "dpm_solver_multistep", "DPMSolverSinglestepScheduler": "dpm_solver_singlestep", "KDPM2DiscreteScheduler": "kdpm2_discrete", "KDPM2AncestralDiscreteScheduler": "kdpm2_ancestral_discrete", "EulerDiscreteScheduler": "euler_discrete", "EulerAncestralDiscreteScheduler": "euler_ancestral_discrete", "HeunDiscreteScheduler": "heun_discrete", "LMSDiscreteScheduler": "lms_discrete", "PNDMScheduler": "pndm", "DEISMultistepScheduler": "deis_multistep", "DPMSolverSDEScheduler": "dpm_solver_sde", "UniPCMultistepScheduler": "uni_pc_multistep" } #################### # # Determine automatically if on CPU or GPU # # CPU will not work on Windows. # device = "cpu" if torch.cuda.is_available(): device = "cuda" PYTORCH_CUDA_ALLOC_CONF = { "max_split_size_mb": 8000 } torch.cuda.max_memory_allocated( device = device ) torch.cuda.empty_cache() if device == "cpu": default_base_model = "sdxl_turbo" #################### # # Determine if running on HuggingFace # try: if (str(os.uname()).find("magicfixeseverything") >= 0): script_being_run_on_hugging_face = 1 except: script_being_run_on_hugging_face = 0 if script_being_run_on_hugging_face == 1: allow_other_model_versions = 0 #################### default_prompt = "" default_negative_prompt = "" default_width = 768 default_height = 768 minimum_width = 64 minimum_height = 64 maximum_width = 2048 # 1024 maximum_height = 2048 # 1024 default_base_model_steps = 50 default_base_model_steps_for_sdxl_turbo = 2 maximum_base_model_steps = 150 # 100 maximum_base_model_steps_for_sdxl_turbo = 25 default_guidance_scale = 7.5 minimum_guidance_scale = 0 maximum_guidance_scale = 30 # Must be greater than 0 guidance_scale_input_slider_steps = 0.25 # # To have the seed be random, set this to: # # "random" # default_seed_value = "random" maximum_seed = 1000000000000000000 add_generation_information_to_image = 1 # If you turn off the refiner it will not be available in the display unless # you select an online configuration option that requires it. enable_refiner = 1 enable_upscaler = 1 # Selected on form as a default? default_refiner_selected = 0 default_upscaler_selected = 0 # Accordion visible on load? # # 0 If selected as default, will be open. Otherwise, closed. # 1 Always starts open default_refiner_accordion_open = 1 default_upscaler_accordion_open = 1 default_refiner_denoise_start = 0.95 minimum_refiner_denoise_start = 0.01 maximum_refiner_denoise_start = 0.99 # Must be greater than 0 refiner_denoise_start_input_slider_steps = 0.01 # Only for SDXL online configuration. We just use base model steps and denoise start normally default_refining_steps_for_online_config_field = 100 maximum_refining_steps_for_online_config_field = 100 # Upscaler Options maximum_upscaler_steps = 150 default_upscaler_steps = 50 # xFormers: # # https://huggingface.co/docs/diffusers/optimization/xformers use_xformers = 1 # Scaled dot product attention (SDPA) is used by default for PyTorch 2.0. To # use default instead, set this to 1. # # https://huggingface.co/docs/diffusers/optimization/torch2.0#scaled-dot-product-attention use_default_attn_processor = 0 display_xformers_usage_in_prompt_info = 1 include_transformers_version_in_prompt_info = 1 display_diffusers_version_in_prompt_info = 1 display_default_attn_processor_usage_in_prompt_info = 1 # You can't select both sequential and model cpu offloading. If you select # both, model cpu offloading will be used. use_sequential_cpu_offload_for_base_model = 1 use_sequential_cpu_offload_for_refiner = 1 use_sequential_cpu_offload_for_upscaler = 1 use_model_cpu_offload_for_base_model = 0 use_model_cpu_offload_for_refiner = 0 use_model_cpu_offload_for_upscaler = 0 # Doesn't work on Windows as of writing this use_torch_compile_for_base_model = 0 use_torch_compile_for_refiner = 0 use_torch_compile_for_upscaler = 0 if default_base_model == "sdxl": # SDXL default_width = 1024 default_height = 1024 #elif default_base_model == "photoreal": # PhotoReal #elif default_base_model == "sdxl_turbo": # SDXL Turbo #elif default_base_model == "sd_1_5_runwayml": # SD 1.5 # Must be multiple of 8 width_and_height_input_slider_steps = 8 maximum_prompt_characer_count = 1000 opening_html = "" ending_html = "" max_queue_size = max_queue_size_if_torch if device == "cpu": if script_being_run_on_hugging_face == 1: opening_html = "This app is extremely slow. Do not use it yet. This app is not running on a GPU. The first time it loads after the space is rebuilt it might take 10 minutes to generate a SDXL Turbo image. It may take around 3 minutes after that point to do two steps. (with no refining or upscaling) For other models, it may take an hour to create a single image. Want apps that work fast? Use these which this app is based on: Stable Diffusion XL, PhotoReal with SDXL 1.0 Refiner and SDXL Turbo Unofficial Demo. This app is designed to give more options, but it's too slow to operate and test on a CPU. There are some features that are either not available, or more limited, in the online version of this app. (such as smaller allowed image dimensions and less steps allowed) This app is still in development. A future addition will be the ability to create a link like this." else: opening_html = "This app is currently running on a CPU. If you have a NVIDIA graphics card, make sure you have torch installed so that you can use your GPU to create imagery. If you don't, it will work extremely slowly." max_queue_size = max_queue_size_if_cpu ending_html += """Spaghetti AI Logo Tokens are not individual characters. If the prompt length is too long, the display will notify you what part of the prompt wasn't used. Changing just the image dimensions alone will change the image generated. For some models, trying to make a large image, such as 1024x1024, may add extra people and come out worse than using smaller dimensions. The original script for this app was written by Manjushri.""" refiner_on_text = "Refiner is on. " refiner_off_text = "Refiner is off. " upscaler_on_text = "Upscaler is on. " upscaler_off_text = "Upscaler is off. " number_of_reserved_tokens = 2 generate_image_button_normal_text = "Generate Image" generate_image_button_in_progress_text = "Generating..." cancel_image_button_text = "Cancel" cancel_image_button_in_progress_text = "Cancelling..." gradio_image_component_height = 300 gradio_image_gallery_component_height = 350 canceled_image_in_queue_message = "Due to how the queue works in this app, you need to reload the page after canceling an image. Otherwise, you will not be able to generate another image until you reach the position in the queue you were in originally. At that time, the button to generate an image will appear again." canceled_image_in_process_of_being_generated = "
Image generation will be canceled once the current step completes.
" ############################################################################### ############################################################################### # # # # End Configurations # # # ############################################################################### ############################################################################### if default_add_seed_into_pipe == 1: default_use_torch_manual_seed_but_do_not_add_to_pipe = 0 if script_being_run_on_hugging_face == 1: # If on HuggingFace, I change some things. use_custom_hugging_face_cache_dir = 0 auto_save_imagery = 0 show_messages_in_modal_on_page = 0 show_messages_in_command_prompt = 1 only_use_local_files = 0 if device == "cpu": # If on CPU at HuggingFace, I reduce what is available amd do other # things. use_torch_compile_for_base_model = 0 use_torch_compile_for_refiner = 0 use_torch_compile_for_upscaler = 0 show_image_creation_progress_log = 1 minimum_width = 256 minimum_height = 256 maximum_width = 768 maximum_height = 768 minimum_guidance_scale = 1 maximum_guidance_scale = 15 maximum_base_model_steps = 30 maximum_base_model_steps_for_sdxl_turbo = 5 minimum_refiner_denoise_start = 0.70 maximum_upscaler_steps = 15 default_upscaler_steps = 10 ending_html = """ If you would like to download this app to run offline on a Windows computer that has a NVIDIA graphics card, click here to download it. """ + ending_html if default_width < minimum_width: default_width = minimum_width if default_height < minimum_height: default_height = minimum_height if default_width > maximum_width: default_width = maximum_width if default_height > maximum_height: default_height = maximum_height if default_base_model_steps > maximum_base_model_steps: default_base_model_steps = maximum_base_model_steps if default_base_model_steps_for_sdxl_turbo > maximum_base_model_steps_for_sdxl_turbo: default_base_model_steps_for_sdxl_turbo = maximum_base_model_steps_for_sdxl_turbo if default_guidance_scale < minimum_guidance_scale: default_guidance_scale = minimum_guidance_scale if default_guidance_scale > maximum_guidance_scale: default_guidance_scale = maximum_guidance_scale if default_upscaler_steps > maximum_upscaler_steps: default_upscaler_steps = maximum_upscaler_steps only_use_local_files_bool = False if only_use_local_files == 1: only_use_local_files_bool = True if allow_other_model_versions == 0: base_model_array = [ "sdxl", "photoreal", "sdxl_turbo" ] base_model_object_of_model_configuration_arrays = { "sdxl": [ "sdxl_default" ], "photoreal": [ "photoreal_default" ], "sdxl_turbo": [ "sdxl_turbo_default" ] } base_model_model_configuration_defaults_object = { "sdxl": "sdxl_default", "photoreal": "photoreal_default", "sdxl_turbo": "sdxl_turbo_default" } hugging_face_hub_is_offline = 0 if script_being_run_on_hugging_face == 0: if ( ("HF_HUB_OFFLINE" in os.environ) and (int(os.environ["HF_HUB_OFFLINE"]) == 1) ): hugging_face_hub_is_offline = 1 only_use_local_files = 1 if suppress_hugging_face_hub_offline_status == 1: if hugging_face_hub_is_offline == 0: print ("Note: The Hugging Face cache directory does not automatically delete older data. Over time, it could eventually grow to use all the space on the drive it is on. You either need to manually clean out the folder occasionally or see Instructons.txt on how to not automatically update data once you have downloaded everything you need.") else: print ("You are working offline. Data will not be downloaded. See \"ai_image_creation.bat\" or \"Instructions.txt\" for more info.") saved_images_dir = main_dir + "/" + saved_images_folder_name hugging_face_cache_dir = main_dir + "/" + cache_directory_folder_name if not os.path.exists(hugging_face_cache_dir): os.makedirs(hugging_face_cache_dir) if auto_save_imagery == 1: from datetime import datetime if device == "cpu": use_sequential_cpu_offload_for_base_model = 0 use_sequential_cpu_offload_for_refiner = 0 use_sequential_cpu_offload_for_upscaler = 0 use_model_cpu_offload_for_base_model = 0 use_model_cpu_offload_for_refiner = 0 use_model_cpu_offload_for_upscaler = 0 use_xformers = 0 if ( (use_sequential_cpu_offload_for_base_model == 1) and (use_model_cpu_offload_for_base_model == 1) ): use_sequential_cpu_offload_for_base_model = 0 if ( (use_sequential_cpu_offload_for_refiner == 1) and (use_model_cpu_offload_for_refiner == 1) ): use_sequential_cpu_offload_for_refiner = 0 if ( (use_sequential_cpu_offload_for_upscaler == 1) and (use_model_cpu_offload_for_upscaler == 1) ): use_sequential_cpu_offload_for_upscaler = 0 def error_function( text_message ): print (text_message) raise Exception(text_message) additional_prompt_info_html = "" if auto_save_imagery == 1: additional_prompt_info_html = " The image, and a text file with generation information, will be saved automatically." if use_xformers == 1: from xformers.ops import MemoryEfficientAttentionFlashAttentionOp if use_default_attn_processor == 1: from diffusers.models.attention_processor import AttnProcessor if ( default_base_model and (default_base_model in base_model_object_of_model_configuration_arrays) and (default_base_model in base_model_model_configuration_defaults_object) ): default_model_configuration = base_model_model_configuration_defaults_object[default_base_model] if default_model_configuration in model_configuration_names_object: default_model_configuration_choices_array = [] for this_model_configuration in base_model_object_of_model_configuration_arrays[default_base_model]: if this_model_configuration in model_configuration_names_object: default_model_configuration_choices_array.append( model_configuration_names_object[this_model_configuration] ) else: error_function("A default model version must be properly named in the code.") else: error_function("A default model version must be properly configured in the code.") else: error_function("A default base model must be properly configured in the code.") default_base_model_nicely_named_value = base_model_names_object[default_base_model] default_model_configuration_nicely_named_value = model_configuration_names_object[default_model_configuration] if not ( default_scheduler and default_scheduler in scheduler_long_names_object ): error_function("A default scheduler must be properly configured in the code.") default_scheduler_nicely_named_value = scheduler_long_names_object[default_scheduler] if enable_refiner != 1: default_refiner_selected = 0 if enable_upscaler != 1: default_upscaler_selected = 0 default_refine_option = "No" if default_refiner_selected == 1: default_refine_option = "Yes" default_upscale_option = "No" if default_upscaler_selected == 1: default_upscale_option = "Yes" default_refiner_and_upscaler_status_text = "" default_use_denoising_start_in_base_model_when_using_refiner_is_selected = False if default_use_denoising_start_in_base_model_when_using_refiner == 1: default_use_denoising_start_in_base_model_when_using_refiner_is_selected = True default_base_model_output_to_refiner_is_in_latent_space_is_selected = False if default_base_model_output_to_refiner_is_in_latent_space == 1: default_base_model_output_to_refiner_is_in_latent_space_is_selected = True refiner_accordion_visible = True if enable_refiner != 1: refiner_accordion_visible = False refiner_accordion_open = False if default_refiner_accordion_open == 1: refiner_accordion_open = True refiner_group_visible = False if enable_refiner == 1: refiner_group_visible = True if default_refiner_selected == 1: default_refiner_and_upscaler_status_text += refiner_on_text else: default_refiner_and_upscaler_status_text += refiner_off_text upscaler_accordion_open = False if ( (default_upscaler_selected == 1) or (default_upscaler_accordion_open == 1) ): upscaler_accordion_open = True upscaler_group_visible = False if enable_upscaler == 1: upscaler_group_visible = True if default_upscaler_selected == 1: default_refiner_and_upscaler_status_text += upscaler_on_text else: default_refiner_and_upscaler_status_text += upscaler_off_text default_negative_prompt_field_row_visibility = True default_negative_prompt_for_sdxl_turbo_field_row_visibility = False default_base_model_steps_field_row_visibility = True default_base_model_steps_field_for_sdxl_turbo_field_row_visibility = False default_guidance_scale_field_row_visibility = True default_guidance_scale_for_sdxl_turbo_field_row_visibility = False if default_base_model == "sdxl_turbo": default_negative_prompt_field_row_visibility = False default_negative_prompt_for_sdxl_turbo_field_row_visibility = True default_base_model_steps_field_row_visibility = False default_base_model_steps_field_for_sdxl_turbo_field_row_visibility = True default_guidance_scale_field_row_visibility = False default_guidance_scale_for_sdxl_turbo_field_row_visibility = True default_refining_use_denoising_start_in_base_model_when_using_refiner_field_row_visibility = True if default_base_model in base_models_not_supporting_denoising_end_for_base_model_object: default_refining_use_denoising_start_in_base_model_when_using_refiner_field_row_visibility = False default_add_seed_into_pipe_is_selected = False if default_add_seed_into_pipe == 1: default_add_seed_into_pipe_is_selected = True default_use_torch_manual_seed_but_do_not_add_to_pipe_is_selected = False if default_use_torch_manual_seed_but_do_not_add_to_pipe == 1: default_use_torch_manual_seed_but_do_not_add_to_pipe_is_selected = True default_save_base_image_when_using_refiner_is_selected = False if default_save_base_image_when_using_refiner == 1: default_save_base_image_when_using_refiner_is_selected = True default_save_refined_image_when_using_upscaler_is_selected = False if default_save_base_image_when_using_refiner == 1: default_save_refined_image_when_using_upscaler_is_selected = True default_base_model_choices_array = [] stored_model_configuration_names_object = {} for this_base_model in base_model_array: default_base_model_choices_array.append( base_model_names_object[this_base_model] ) stored_model_configuration = base_model_model_configuration_defaults_object[this_base_model] stored_model_configuration_names_object[this_base_model] = model_configuration_names_object[stored_model_configuration] default_scheduler_choices_array = [] for this_scheduler in schedulers_array: default_scheduler_choices_array.append( scheduler_long_names_object[this_scheduler] ) if enable_image_preview != 1: use_image_preview = 0 make_seed_selection_a_textbox = 1 if maximum_seed <= 9007199254740992: make_seed_selection_a_textbox = 0 current_preview_image = "" current_preview_image_user_id = 0 current_image_generation_id_in_progress = 0 cancel_image_generation_ids_object = {} cancel_image_generation_times_object = {} seconds_to_store_cancellations_in_cancel_image_generation_times_object = 86400 ############################################################################### ############################################################################### # # # # # # # Functions # # # # # # ############################################################################### ############################################################################### ##################### # # Rounded Number # # A better, and seemingly more accurate, way to round. # # https://realpython.com/python-rounding/ ##################### def rounded_number(n, decimals=0): n = float(n) multiplier = 10**decimals rounded_abs = (floor(abs(n) * multiplier + 0.5) / multiplier) rounded_value = round(copysign(rounded_abs, n), decimals) return rounded_value ##################### # # Rounded Number # # Format number to a certain number of decimal places and output it to a string. # # https://stackoverflow.com/questions/1995615/how-can-i-format-a-decimal-to-always-show-2-decimal-places ##################### def formatted_number(n, decimals=0): rounded_value = rounded_number(n, decimals) formatted_value = '{:.{prec}f}'.format(rounded_value, prec=decimals) return formatted_value ##################### # # Show Message # # Display message to user in model on web form and/or command prompt. # ##################### def generate_random_seed(): maximum_seed_for_random = maximum_seed if maximum_seed_for_random > 9007199254740992: # If above this number, seeds may not be able to be entered into # slider properly. maximum_seed_for_random = 9007199254740992 actual_seed = int(random.randrange(0, 10**len(str(maximum_seed_for_random)))) return actual_seed ##################### # # Show Message # # Display message to user in model on web form and/or command prompt. # ##################### def show_message( message_to_display ): if show_messages_in_command_prompt == 1: print (message_to_display) if show_messages_in_modal_on_page == 1: gr.Info(message_to_display) ##################### # # Nice Elapsed Time # # Formatted nicely from seconds. # ##################### def nice_elapsed_time( seconds ): # Google AI Code hours = seconds // 3600 minutes = (seconds % 3600) // 60 seconds = seconds % 60 if hours > 0: hours_text = "hr" if hours > 1: hours_text = "hrs" time_html = str(int(hours)) + " " + hours_text + ". " + str(int(minutes)) + " min. " + str(round(seconds, 1)) + " sec." elif minutes > 0: time_html = str(int(minutes)) + " min. " + str(round(seconds, 1)) + " sec." else: time_html = str(round(seconds, 2)) + " sec." return time_html ##################### # # Base Model Valid # # Return True if valid. # ##################### def base_model_valid(base_model_name_value): try: base_model_name_value_str = str(base_model_name_value).lower() if ( (base_model_name_value_str in base_model_object_of_model_configuration_arrays) and (base_model_name_value_str in base_model_model_configuration_defaults_object) ): return True else: return False except ValueError: return False ##################### # # Model Configuration Valid # # Return True if valid. # ##################### def model_configuration_valid( base_model_name_value, model_configuration_name_value ): try: base_model_name_value_str = str(base_model_name_value).lower() model_configuration_name_value_str = str(model_configuration_name_value).lower() for this_base_model in base_model_array: for this_model_configuration in base_model_object_of_model_configuration_arrays[this_base_model]: if ( (base_model_name_value_str == this_base_model) and (model_configuration_name_value_str == this_model_configuration) ): return True return False except ValueError: return False ##################### # # Prompt Valid # # Return True if valid. # ##################### def prompt_valid(prompt_field): try: prompt_field_str = str(prompt_field) if len(prompt_field_str) <= maximum_prompt_characer_count: return True else: return False except ValueError: return False ##################### # # Negative Prompt Valid # # Return True if valid. # ##################### def negative_prompt_valid(negative_prompt_field): try: negative_prompt_field_str = str(negative_prompt_field) if len(negative_prompt_field_str) <= maximum_prompt_characer_count: return True else: return False except ValueError: return False ##################### # # Scheduler/Sampler Valid # # Return True if valid. # ##################### def scheduler_valid(scheduler_field): try: scheduler_str = str(scheduler_field).lower() if scheduler_str in scheduler_long_names_object: return True else: return False except ValueError: return False ##################### # # Width Valid # # Return True if valid. # ##################### def width_valid(width_num_str): try: width_num = int(width_num_str) if ( (width_num >= int(minimum_width)) and (width_num <= int(maximum_width)) and (width_num % int(width_and_height_input_slider_steps)) == 0 ): return True else: return False except ValueError: return False ##################### # # Height Valid # # Return True if valid. # ##################### def height_valid(height_num_str): try: height_num = int(height_num_str) if ( (height_num >= int(minimum_height)) and (height_num <= int(maximum_height)) and (height_num % int(width_and_height_input_slider_steps)) == 0 ): return True else: return False except ValueError: return False ##################### # # Guidance Scale Valid # # Return True if valid. # ##################### def guidance_scale_valid(guidance_scale_str): try: guidance_scale_num = float(guidance_scale_str) guidance_scale_num_times_100 = (guidance_scale_num * 100) guidance_scale_num_times_100_with_int = int(guidance_scale_num_times_100) guidance_scale_input_slider_steps_times_100 = (float(guidance_scale_input_slider_steps) * 100) if ( (guidance_scale_num >= float(minimum_guidance_scale)) and (guidance_scale_num <= float(maximum_guidance_scale)) and (guidance_scale_num_times_100 == guidance_scale_num_times_100_with_int) and ((guidance_scale_num_times_100 % guidance_scale_input_slider_steps_times_100) == 0) ): return True else: return False except ValueError: return False ##################### # # Steps Valid # # Return True if valid. # ##################### def steps_valid( steps_num_str, base_model_name_value ): try: steps_num = int(steps_num_str) base_model_name_value_str = str(base_model_name_value).lower() if steps_num > 0: if (base_model_name_value_str == "sdxl_turbo"): if steps_num <= int(maximum_base_model_steps_for_sdxl_turbo): return True else: if steps_num <= int(maximum_base_model_steps): return True return False except ValueError: return False ##################### # # Seed Valid # # Return True if valid. # ##################### def seed_valid( seed_num_str ): try: seed_num = int(seed_num_str) if ( (seed_num >= 0) and (seed_num <= int(maximum_seed)) ): return True else: return False except ValueError: return False ##################### # # Refiner Denoise Start # # Return True if valid. # ##################### def refiner_denoise_start_valid( refiner_denoise_start_str ): try: refiner_denoise_start_num = float(refiner_denoise_start_str) refiner_denoise_start_num_times_100 = (refiner_denoise_start_num * 100) refiner_denoise_start_num_times_100_with_int = int(refiner_denoise_start_num_times_100) refiner_denoise_start_input_slider_steps_times_100 = (float(refiner_denoise_start_input_slider_steps) * 100) if ( (refiner_denoise_start_num >= float(minimum_refiner_denoise_start)) and (refiner_denoise_start_num <= float(maximum_refiner_denoise_start)) and (refiner_denoise_start_num_times_100 == refiner_denoise_start_num_times_100_with_int) and ((refiner_denoise_start_num_times_100 % refiner_denoise_start_input_slider_steps_times_100) == 0) ): return True else: return False except ValueError: return False ##################### # # Refiner Steps # # Return True if valid. # ##################### def refining_steps_valid( refining_steps_num_str ): try: refining_steps_num = int(refining_steps_num_str) if ( (refining_steps_num > 0) and (refining_steps_num <= int(maximum_refining_steps_for_online_config_field)) ): return True else: return False except ValueError: return False ##################### # # Upscaler Steps # # Return True if valid. # ##################### def upscaling_steps_valid( upscaling_steps_num_str ): try: upscaling_steps_num = int(upscaling_steps_num_str) if ( (upscaling_steps_num > 0) and (upscaling_steps_num <= int(maximum_upscaler_steps)) ): return True else: return False except ValueError: return False ##################### # # Numerical Bool # # Return 1 for anything that is True/Yes/1. Everything else is False. # ##################### def numerical_bool( original_value ): new_value = 0 if ( (original_value == 1) or (original_value == "Yes") or (original_value == "True") or (original_value == True) ): new_value = 1 return new_value ##################### # # Truncate Prompt # # Truncate a prompt. Get the actual prompt that will be used and save the # part of the prompt that will not be used. # ##################### def truncate_prompt ( pipe, existing_prompt_text ): # Only 77 tokens are allowed in the prompt. 2 are reserved, meaning it is # truncated to 75. This happens automatically, but we want to tell people # that tokenizer = pipe.tokenizer max_token_length_of_model = pipe.tokenizer.model_max_length - number_of_reserved_tokens prompt_text_words_array = existing_prompt_text.split(" ") prompt_text_words_array_length = len(prompt_text_words_array) prompt_text_words_index = 0 prompt_text_substring = "" prompt_text_not_used_substring = "" for prompt_text_word in prompt_text_words_array: prompt_text_words_index += 1 substring_to_test = prompt_text_substring if prompt_text_words_index > 1: substring_to_test += " " substring_to_test += prompt_text_word token_length_of_substring_to_test = len(tokenizer.tokenize(substring_to_test)) if token_length_of_substring_to_test > max_token_length_of_model: prompt_text_not_used_substring += prompt_text_word + " " else: prompt_text_substring = substring_to_test return ( prompt_text_substring, prompt_text_not_used_substring ) ##################### # # Construct Pipe # # Prepare the base model. # ##################### def construct_pipe ( base_model_name_value, model_configuration_name_value ): if device == "cuda": torch.cuda.empty_cache() base_model_kwargs = {} if ( (base_model_name_value == "sdxl") or (base_model_name_value == "photoreal") or (base_model_name_value == "sdxl_turbo") or (base_model_name_value == "sd_1_5_runwayml") ): base_model_kwargs["use_safetensors"] = True if use_safety_checker == 0: if ( (base_model_name_value == "photoreal") or (base_model_name_value == "sd_1_5_runwayml") ): base_model_kwargs = { "safety_checker": None, "requires_safety_checker": False } if device == "cuda": if ( (base_model_name_value == "sdxl") or (base_model_name_value == "sdxl_turbo") or (base_model_name_value == "sd_1_5_runwayml") ): base_model_kwargs["variant"] = "fp16" base_model_kwargs["torch_dtype"] = torch.float16 if use_custom_hugging_face_cache_dir == 1: base_model_kwargs["cache_dir"] = hugging_face_cache_dir pipe = DiffusionPipeline.from_pretrained( pretrained_model_name_or_path = model_configuration_links_object[model_configuration_name_value], local_files_only = only_use_local_files_bool, **base_model_kwargs ) if use_model_cpu_offload_for_base_model == 1: pipe.enable_model_cpu_offload() if use_xformers == 1: pipe.enable_xformers_memory_efficient_attention() pipe = pipe.to(device) if use_sequential_cpu_offload_for_base_model == 1: pipe.enable_sequential_cpu_offload() if use_default_attn_processor == 1: pipe.unet.set_default_attn_processor() if use_torch_compile_for_base_model == 1: pipe.unet = torch.compile( pipe.unet, mode = "reduce-overhead", fullgraph = True ) if device == "cuda": torch.cuda.empty_cache() return ( pipe ) ##################### # # Configure Scheduler # ##################### def configure_scheduler ( pipe, scheduler_value ): scheduler_config = pipe.scheduler.config scheduler = scheduler_value if scheduler_value == "model_default": scheduler_name = pipe.scheduler.config._class_name if scheduler_name in scheduler_name_to_identifier_in_app_object: scheduler = scheduler_name_to_identifier_in_app_object[scheduler_name] scheduler_used = scheduler if scheduler == "ddim": from diffusers import DDIMScheduler pipe.scheduler = DDIMScheduler.from_config(scheduler_config) elif scheduler == "ddpm": from diffusers import DDPMScheduler pipe.scheduler = DDPMScheduler.from_config(scheduler_config) elif scheduler == "dpm_solver_multistep": from diffusers import DPMSolverMultistepScheduler pipe.scheduler = DPMSolverMultistepScheduler.from_config(scheduler_config) elif scheduler == "dpm_solver_multistep_karras_sigmas_true": new_scheduler_config = dict(pipe.scheduler.config) new_scheduler_config.update({"use_karras_sigmas": True}) from diffusers import DPMSolverMultistepScheduler pipe.scheduler = DPMSolverMultistepScheduler.from_config(new_scheduler_config) elif scheduler == "dpm_solver_multistep_algorithm_type_sde-dpmsolver_pp": new_scheduler_config = dict(pipe.scheduler.config) new_scheduler_config.update({"algorithm_type": "sde-dpmsolver++"}) from diffusers import DPMSolverMultistepScheduler pipe.scheduler = DPMSolverMultistepScheduler.from_config(new_scheduler_config) elif scheduler == "dpm_solver_multistep_karras_sigmas_true_algorithm_type_sde-dpmsolver_pp": new_scheduler_config = dict(pipe.scheduler.config) new_scheduler_config.update({"use_karras_sigmas": True}) new_scheduler_config.update({"algorithm_type": "sde-dpmsolver++"}) from diffusers import DPMSolverMultistepScheduler pipe.scheduler = DPMSolverMultistepScheduler.from_config(new_scheduler_config) elif scheduler == "dpm_solver_singlestep": from diffusers import DPMSolverSinglestepScheduler pipe.scheduler = DPMSolverSinglestepScheduler.from_config(scheduler_config) elif scheduler == "dpm_solver_singlestep_karras_sigmas_true": new_scheduler_config = dict(pipe.scheduler.config) new_scheduler_config.update({"use_karras_sigmas": True}) from diffusers import DPMSolverSinglestepScheduler pipe.scheduler = DPMSolverSinglestepScheduler.from_config(new_scheduler_config) elif scheduler == "kdpm2_discrete": from diffusers import KDPM2DiscreteScheduler pipe.scheduler = KDPM2DiscreteScheduler.from_config(scheduler_config) elif scheduler == "kdpm2_discrete_karras_sigmas_true": new_scheduler_config = dict(pipe.scheduler.config) new_scheduler_config.update({"use_karras_sigmas": True}) from diffusers import KDPM2DiscreteScheduler pipe.scheduler = KDPM2DiscreteScheduler.from_config(new_scheduler_config) elif scheduler == "kdpm2_ancestral_discrete": from diffusers import KDPM2AncestralDiscreteScheduler pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(scheduler_config) elif scheduler == "kdpm2_ancestral_discrete_karras_sigmas_true": new_scheduler_config = dict(pipe.scheduler.config) new_scheduler_config.update({"use_karras_sigmas": True}) from diffusers import KDPM2AncestralDiscreteScheduler pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(new_scheduler_config) elif scheduler == "euler_discrete": from diffusers import EulerDiscreteScheduler pipe.scheduler = EulerDiscreteScheduler.from_config(scheduler_config) elif scheduler == "euler_ancestral_discrete": from diffusers import EulerAncestralDiscreteScheduler pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(scheduler_config) elif scheduler == "heun_discrete": from diffusers import HeunDiscreteScheduler pipe.scheduler = HeunDiscreteScheduler.from_config(scheduler_config) elif scheduler == "lms_discrete": from diffusers import LMSDiscreteScheduler pipe.scheduler = LMSDiscreteScheduler.from_config(scheduler_config) elif scheduler == "lms_discrete_karras_sigmas_true": new_scheduler_config = dict(pipe.scheduler.config) new_scheduler_config.update({"use_karras_sigmas": True}) from diffusers import LMSDiscreteScheduler pipe.scheduler = LMSDiscreteScheduler.from_config(new_scheduler_config) elif scheduler == "pndm": from diffusers import PNDMScheduler pipe.scheduler = PNDMScheduler.from_config(scheduler_config) elif scheduler == "pndm_skip_prk_steps_true": new_scheduler_config = dict(pipe.scheduler.config) new_scheduler_config.update({"skip_prk_steps": True}) from diffusers import PNDMScheduler pipe.scheduler = PNDMScheduler.from_config(new_scheduler_config) elif scheduler == "deis_multistep": from diffusers import DEISMultistepScheduler pipe.scheduler = DEISMultistepScheduler.from_config(scheduler_config) elif scheduler == "dpm_solver_sde": from diffusers import DPMSolverSDEScheduler pipe.scheduler = DPMSolverSDEScheduler.from_config(scheduler_config) elif scheduler == "uni_pc_multistep": from diffusers import UniPCMultistepScheduler pipe.scheduler = UniPCMultistepScheduler.from_config(scheduler_config) else: from diffusers import PNDMScheduler pipe.scheduler = PNDMScheduler.from_config(scheduler_config) scheduler_used = "pndm" return ( scheduler_used ) ##################### # # Construct Refiner # # Prepare the refiner. # ##################### def construct_refiner (): refiner_kwargs = { "use_safetensors": True } if device == "cuda": refiner_kwargs["variant"] = "fp16" refiner_kwargs["torch_dtype"] = torch.float16 if use_custom_hugging_face_cache_dir == 1: refiner_kwargs["cache_dir"] = hugging_face_cache_dir refiner = DiffusionPipeline.from_pretrained( pretrained_model_name_or_path = hugging_face_refiner_partial_path, local_files_only = only_use_local_files_bool, **refiner_kwargs ) if use_model_cpu_offload_for_refiner == 1: refiner.enable_model_cpu_offload() if use_xformers == 1: refiner.enable_xformers_memory_efficient_attention() refiner = refiner.to(device) if use_sequential_cpu_offload_for_refiner == 1: refiner.enable_sequential_cpu_offload() if use_default_attn_processor == 1: refiner.unet.set_default_attn_processor() if use_torch_compile_for_refiner == 1: refiner.unet = torch.compile( refiner.unet, mode = "reduce-overhead", fullgraph = True ) if device == "cuda": torch.cuda.empty_cache() return ( refiner ) ##################### # # Construct Upscaler # # Prepare the upscaler. # ##################### def construct_upscaler (): upscaler_kwargs = { "use_safetensors": True } if device == "cuda": upscaler_kwargs["torch_dtype"] = torch.float16 if use_custom_hugging_face_cache_dir == 1: upscaler_kwargs["cache_dir"] = hugging_face_cache_dir upscaler = DiffusionPipeline.from_pretrained( pretrained_model_name_or_path = hugging_face_upscaler_partial_path, local_files_only = only_use_local_files_bool, **upscaler_kwargs ) if use_model_cpu_offload_for_upscaler == 1: upscaler.enable_model_cpu_offload() if use_xformers == 1: upscaler.enable_xformers_memory_efficient_attention() upscaler = upscaler.to(device) if use_sequential_cpu_offload_for_upscaler == 1: upscaler.enable_sequential_cpu_offload() if use_default_attn_processor == 1: upscaler.unet.set_default_attn_processor() if use_torch_compile_for_refiner == 1: upscaler.unet = torch.compile( upscaler.unet, mode = "reduce-overhead", fullgraph = True ) if device == "cuda": torch.cuda.empty_cache() return ( upscaler ) ##################### # # Update Prompt Info From Gallery # # If you select an image in the image gallery, display the prompt # information for that image. # ##################### def update_prompt_info_from_gallery ( gallery_data: gr.SelectData, image_gallery_array_state_value ): gallery_data_index = gallery_data.index output_image_gallery_field_update = gr.Gallery( selected_index = gallery_data_index ) output_text_field_update = image_gallery_array_state_value[gallery_data_index] return { output_image_gallery_field: output_image_gallery_field_update, output_text_field: output_text_field_update } ##################### # # Save Image File # # If configured to do so. # ##################### def save_image_file ( saved_image_path_and_file, image_to_return, add_generation_information_to_image, info_to_save_in_image ): if add_generation_information_to_image == 1: from PIL.PngImagePlugin import PngInfo saved_image_metadata = PngInfo() saved_image_metadata.add_text( "parameters", info_to_save_in_image ) image_to_return_file = image_to_return.save( saved_image_path_and_file, pnginfo = saved_image_metadata ) else: image_to_return_file = image_to_return.save( saved_image_path_and_file ) ##################### # # Create Image Generation Information # # This information will be displayed on the page and within the image too # if you were to open it in a text file. # ##################### def create_image_generation_information ( base_model_name_value, model_configuration_name_value, scheduler_used, scheduler_value, prompt_text, prompt_text_not_used_substring, negative_prompt_text, negative_prompt_text_not_used_substring, image_width, image_height, actual_seed, add_seed_into_pipe, guidance_scale, base_model_steps, display_xformers_usage_in_prompt_info, display_default_attn_processor_usage_in_prompt_info, display_diffusers_version_in_prompt_info, use_refiner, refiner_error, refining_denoise_start_field_value, denoising_end_applicable, refining_use_denoising_start_in_base_model_when_using_refiner_field_value, refining_base_model_output_to_refiner_is_in_latent_space_field_value, refining_steps_for_older_configuration_field_value, use_upscaler, upscaler_error, upscaling_steps, upscaled_image_width, upscaled_image_height, current_actual_total_base_model_steps, current_actual_total_refiner_steps, current_actual_total_upscaler_steps, generation_start_time, image_has_been_canceled, which_image ): # # # # Prompt Information # # # info_about_prompt_lines_array = [] if image_has_been_canceled == 1: info_about_prompt_lines_array.extend([ "Image was canceled before completion. Some details below may not be accurate." ]) if refiner_error == 1: info_about_prompt_lines_array.extend([ "Refiner Error: An error occurred in the refining progress and was skipped. Some details below will not be accurate." ]) if upscaler_error == 1: info_about_prompt_lines_array.extend([ "Upscaler Error: An error occurred in the upscaling progress and was skipped. Some details below will not be accurate." ]) info_about_prompt_lines_array.extend([ "Prompt: " + prompt_text ]) if len(negative_prompt_text) > 0: info_about_prompt_lines_array.extend([ "Negative Prompt: " + negative_prompt_text ]) dimensions_title = "Dimensions" if use_upscaler == 1: dimensions_title = "Original Dimensions" info_about_prompt_lines_array.extend([ dimensions_title + ": " + str(image_width) + "x" + str(image_height) + " px" ]) if use_upscaler == 1: info_about_prompt_lines_array.extend([ "Upscaled Dimensions: " + str(upscaled_image_width) + "x" + str(upscaled_image_height) + " px" ]) info_about_prompt_lines_array.extend([ "Seed: " + str(actual_seed) ]) nice_seed_added_to_generation = "No" if add_seed_into_pipe == 1: nice_seed_added_to_generation = "Yes" info_about_prompt_lines_array.extend([ "Seed added to generation? " + nice_seed_added_to_generation ]) if int(guidance_scale) > 0: info_about_prompt_lines_array.extend([ "Guidance Scale: " + str(guidance_scale) ]) if ( (image_has_been_canceled == 0) or (base_model_steps == current_actual_total_base_model_steps) ): info_about_prompt_lines_array.extend([ "Steps: " + str(base_model_steps) ]) else: info_about_prompt_lines_array.extend([ "Selected Steps: " + str(base_model_steps), "Actual Base Model Steps: " + str(current_actual_total_base_model_steps) ]) nice_model_name = base_model_names_object[base_model_name_value] + " (" + model_configuration_links_object[model_configuration_name_value] + ")" nice_scheduler_name = scheduler_short_names_object[scheduler_used] if scheduler_value == "model_default": nice_scheduler_name += " (model default)" info_about_prompt_lines_array.extend([ "Model: " + nice_model_name, "Scheduler/Sampler: " + nice_scheduler_name ]) if ( (use_refiner == 1) or (refiner_error == 1) ): refiner_usage_text = "Yes" if refiner_error == 1: refiner_usage_text = "No (an error prevented it from refining)\nHowever, selected refining details are still included." info_about_prompt_lines_array.extend([ "Refiner Used? " + refiner_usage_text ]) nice_refiner_denoise_start = str(refining_denoise_start_field_value) if denoising_end_applicable == 1: if refining_use_denoising_start_in_base_model_when_using_refiner_field_value == 1: info_about_prompt_lines_array.extend([ "Set \"denoising_end\" in base model generation? Yes", "Base model denoise end %: " + nice_refiner_denoise_start ]) if current_actual_total_base_model_steps > 0: info_about_prompt_lines_array.extend([ "Actual Base Model Steps: " + formatted_number(current_actual_total_base_model_steps) ]) else: info_about_prompt_lines_array.extend([ "Set \"denoising_end\" in base model generation? No", ]) info_about_prompt_lines_array.extend([ "Refiner denoise start %: " + nice_refiner_denoise_start ]) if current_actual_total_refiner_steps > 0: info_about_prompt_lines_array.extend([ "Actual Refining Steps: " + formatted_number(current_actual_total_refiner_steps) ]) if refining_base_model_output_to_refiner_is_in_latent_space_field_value == 1: info_about_prompt_lines_array.extend([ "Base model output in latent space before refining? Yes", ]) else: info_about_prompt_lines_array.extend([ "Base model output in latent space before refining? No", ]) if ( (use_upscaler == 1) or (upscaler_error == 1) ): if use_upscaler == 1: info_about_prompt_lines_array.extend([ "Upscaled (2x)? Yes", "Upscaler Steps: " + str(upscaling_steps) ]) elif upscaler_error: info_about_prompt_lines_array.extend([ "Upscaled (2x)? No (an error prevented it from upscaling)", "Selected Upscaler Steps: " + str(upscaling_steps) ]) if log_generation_times == 1: generation_end_time = time.time() generation_time_in_seconds = (generation_end_time - generation_start_time) nice_generation_time = nice_elapsed_time(generation_time_in_seconds) info_about_prompt_lines_array.extend([ "Time: " + nice_generation_time ]) if len(prompt_text_not_used_substring) > 0: info_about_prompt_lines_array.extend([ "End of Prompt Truncated: " + prompt_text_not_used_substring ]) if len(negative_prompt_text_not_used_substring) > 0: info_about_prompt_lines_array.extend([ "End of Negative Prompt Truncated: " + negative_prompt_text_not_used_substring ]) if display_xformers_usage_in_prompt_info == 1: nice_xformers_usage = "No" if use_xformers == 1: nice_xformers_usage = "Yes" if include_transformers_version_in_prompt_info == 1: import transformers nice_xformers_usage += " (version " + str(transformers.__version__) + ")" info_about_prompt_lines_array.extend([ "xFormers Used?: " + nice_xformers_usage ]) if display_default_attn_processor_usage_in_prompt_info == 1: nice_default_attn_processor_usage = "No" if use_default_attn_processor == 1: nice_default_attn_processor_usage = "Yes" info_about_prompt_lines_array.extend([ "Default AttnProcessor Used? " + nice_default_attn_processor_usage ]) if display_diffusers_version_in_prompt_info == 1: try: import diffusers diffusers_version = diffusers.__version__ except: diffusers_version = "" if diffusers_version: info_about_prompt_lines_array.extend([ "Diffusers Version: " + str(diffusers_version) ]) info_about_prompt = "\n".join(info_about_prompt_lines_array) return info_about_prompt ##################### # # Load Image Preview # # If needed, this loads the image preview. # ##################### def load_image_preview ( user_id_state ): user_id_state_value = user_id_state.value # print ("Image preview check run", current_preview_image_user_id, user_id_state_value) if ( (user_id_state_value > 0) and (user_id_state_value == current_preview_image_user_id) ): # print ("Image preview shown") return { output_image_preview_field: current_preview_image[0] } else: return None ##################### # # Before Create Image Function # # This is loaded before the image creation begins. # ##################### def before_create_image_function (): generate_image_button_update = gr.Button( value = generate_image_button_in_progress_text, variant = "secondary", interactive = False ) output_text_field_update = gr.Textbox( visible = False ) prompt_truncated_field_group_update = gr.Group( visible = False ) prompt_truncated_field_update = gr.HTML( value = "" ) negative_prompt_truncated_field_group_update = gr.Group( visible = False ) negative_prompt_truncated_field_update = gr.HTML( value = "" ) error_text_field_update = gr.HTML( value = "", visible = False ) image_generation_id = int(random.randrange(0, 1000000000)) image_generation_id_state_update = gr.State( value = image_generation_id ) before_create_image_object = { generate_image_button: generate_image_button_update, output_text_field: output_text_field_update, prompt_truncated_field_group: prompt_truncated_field_group_update, prompt_truncated_field: prompt_truncated_field_update, negative_prompt_truncated_field_group: negative_prompt_truncated_field_group_update, negative_prompt_truncated_field: negative_prompt_truncated_field_update, error_text_field: error_text_field_update, image_generation_id_state: image_generation_id_state_update } if use_image_preview == 1: output_image_field_update = gr.Image( height = 100 ) output_image_gallery_field_update = gr.Gallery( height = 100 ) output_image_preview_field_row_update = gr.Row( visible = True ) before_create_image_object.update({ output_image_field: output_image_field_update, output_image_preview_field_row: output_image_preview_field_row_update }) if enable_image_generation_cancellation == 1: cancel_image_button_row_update = gr.Row( visible = True ) cancel_image_button_update = gr.Button( interactive = True ) before_create_image_object.update({ cancel_image_button_row: cancel_image_button_row_update, cancel_image_button: cancel_image_button_update }) return before_create_image_object ##################### # # After Create Image Function # # This is loaded once image creation has completed. # ##################### def after_create_image_function (): generate_image_button_update = gr.Button( value = generate_image_button_normal_text, variant = "primary", interactive = True ) output_text_field_update = gr.Textbox( visible = True ) after_create_image_object = { generate_image_button: generate_image_button_update, output_text_field: output_text_field_update } if use_image_preview == 1: output_image_field_update = gr.Image( height = gradio_image_component_height ) output_image_gallery_field_update = gr.Gallery( height = gradio_image_gallery_component_height ) output_image_preview_field_row_update = gr.Row( visible = False ) after_create_image_object.update({ output_image_field: output_image_field_update, output_image_gallery_field: output_image_gallery_field_update, output_image_preview_field_row: output_image_preview_field_row_update }) if enable_image_generation_cancellation == 1: generate_image_button_row_update = gr.Row( visible = True ) cancel_image_button_row_update = gr.Row( visible = False ) cancel_image_button_update = gr.Button( value = cancel_image_button_text, interactive = False ) cancel_image_message_field_row_update = gr.Button( visible = False ) cancel_image_message_field_update = gr.Button( value = "" ) after_create_image_object.update({ generate_image_button_row: generate_image_button_row_update, cancel_image_button_row: cancel_image_button_row_update, cancel_image_button: cancel_image_button_update, cancel_image_message_field_row: cancel_image_message_field_row_update, cancel_image_message_field: cancel_image_message_field_update }) return after_create_image_object ##################### # # Remove From Cancel Object # # Remove user id and generation id once something is canceled. # ##################### def remove_from_cancel_object ( user_id_state_value, image_generation_id_state_value ): if ( (user_id_state_value in cancel_image_generation_ids_object) and (image_generation_id_state_value in cancel_image_generation_ids_object[user_id_state_value]) ): cancel_image_generation_ids_object[user_id_state_value].remove(image_generation_id_state_value) if len(cancel_image_generation_ids_object[user_id_state_value]) == 0: cancel_image_generation_ids_object.pop(user_id_state_value, None) if image_generation_id_state_value in cancel_image_generation_times_object: cancel_image_generation_times_object.pop(image_generation_id_state_value, None) ##################### # # Image Processing Is Canceled # # Return True if image processing has been canceled. # ##################### def image_processing_is_canceled ( user_id_state_value, image_generation_id_state_value ): global cancel_image_generation_ids_object image_processing_was_canceled = False if ( enable_image_generation_cancellation and (user_id_state_value in cancel_image_generation_ids_object) and (image_generation_id_state_value in cancel_image_generation_ids_object[user_id_state_value]) ): image_processing_was_canceled = True return image_processing_was_canceled ##################### # # Create Image from Latents # # # ##################### def create_image_from_latents ( model_to_use, pipe, latents, generator, is_final_image ): if ( (model_to_use == "sdxl") or (model_to_use == "sdxl_turbo") ): # https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py # make sure the VAE is in float32 mode, as it overflows in float16 needs_upcasting = pipe.vae.dtype == torch.float16 and pipe.vae.config.force_upcast if needs_upcasting: pipe.upcast_vae() latents = latents.to(next(iter(pipe.vae.post_quant_conv.parameters())).dtype) image = pipe.vae.decode(latents / pipe.vae.config.scaling_factor, return_dict=False)[0] # cast back to fp16 if needed if needs_upcasting: pipe.vae.to(dtype=torch.float16) if int(is_final_image) == 1: # apply watermark if available if pipe.watermark is not None: image = pipe.watermark.apply_watermark(image) image = pipe.image_processor.postprocess(image, output_type="pil") return image elif ( (model_to_use == "sd_1_5_runwayml") or (model_to_use == "photoreal") ): # https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py image = pipe.vae.decode(latents / pipe.vae.config.scaling_factor, return_dict=False, generator=generator)[0] do_denormalize = [True] * image.shape[0] image = pipe.image_processor.postprocess(image, output_type="pil", do_denormalize=do_denormalize) return image elif model_to_use == "upscaler": # https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py # make sure the VAE is in float32 mode, as it overflows in float16 needs_upcasting = pipe.vae.dtype == torch.float16 and pipe.vae.config.force_upcast if needs_upcasting: pipe.upcast_vae() # Ensure latents are always the same type as the VAE latents = latents.to(next(iter(pipe.vae.post_quant_conv.parameters())).dtype) image = pipe.vae.decode(latents / pipe.vae.config.scaling_factor, return_dict=False)[0] # cast back to fp16 if needed if needs_upcasting: pipe.vae.to(dtype=torch.float16) if int(is_final_image) == 1: # apply watermark if available if pipe.watermark is not None: image = pipe.watermark.apply_watermark(image) do_denormalize = [True] * image.shape[0] image = pipe.image_processor.postprocess(image, output_type="pil", do_denormalize=do_denormalize) return image return "" ##################### # # Create Preview Image # # Add to a global variable to display preview image # ##################### def create_preview_image ( model_to_use, user_id_state, pipe, latents, generator, temporary_extra ): is_final_image = 0 pil_image = create_image_from_latents( model_to_use, pipe, latents, generator, is_final_image ) if pil_image: global current_preview_image global current_preview_image_user_id current_preview_image = pil_image current_preview_image_user_id = user_id_state # For testing # for i, img in enumerate(pil_image): # filename = str(temporary_extra) + "_" + str(i) + ".png" # img.info = {"parameters": "Preview"} # img.save(filename) return None ##################### # # Create Image Function # # This is the main image creation function. # ##################### def create_image_function ( base_model_field_index, prompt_text, negative_prompt_text, scheduler_index, image_width, image_height, guidance_scale, base_model_steps, base_model_steps_field_for_sdxl_turbo, actual_seed, add_seed_into_pipe, use_torch_manual_seed_but_do_not_add_to_pipe_field, refining_selection_field_value, refining_denoise_start_field_value, refining_use_denoising_start_in_base_model_when_using_refiner_field_value, refining_base_model_output_to_refiner_is_in_latent_space_field_value, refining_steps_for_older_configuration_field_value, upscaling_selection_field_value, upscaling_steps, image_gallery_array_state_value, prompt_information_array_state_value, last_model_configuration_name_selected_state_value, last_refiner_name_selected_state_value, last_upscaler_name_selected_state_value, stored_pipe_state, stored_refiner_state, stored_upscaler_state, save_base_image_when_using_refiner_field_value, save_refined_image_when_using_upscaler_field_value, user_id_state, image_generation_id_state, canceled_images_array_state, *model_configuration_dropdown_fields_array, progress = gr.Progress() ): image_generation_id_state_value = image_generation_id_state.value user_id_state_value = user_id_state.value global current_image_generation_id_in_progress current_image_generation_id_in_progress = image_generation_id_state_value global cancel_image_generation_ids_object global cancel_image_generation_times_object global current_preview_image global current_preview_image_user_id current_preview_image = None current_preview_image_user_id = 0 if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value): # User likely canceled while waiting in queue remove_from_cancel_object(user_id_state_value, image_generation_id_state_value) output_text_field_update = gr.Textbox() return { output_text_field: output_text_field_update } image_width = int(image_width) image_height = int(image_height) guidance_scale = float(guidance_scale) base_model_steps = int(base_model_steps) base_model_steps_field_for_sdxl_turbo = int(base_model_steps_field_for_sdxl_turbo) actual_seed = int(actual_seed) refining_denoise_start_field_value = rounded_number(refining_denoise_start_field_value, 2) refining_steps_for_older_configuration_field_value = int(refining_steps_for_older_configuration_field_value) upscaling_steps = int(upscaling_steps) base_model_name_value = base_model_array[base_model_field_index] position_in_array = 0 model_configuration_field_object = {} for model_configuration_field_index in model_configuration_dropdown_fields_array: this_base_model = base_model_array[position_in_array] model_configuration_field_object[this_base_model] = model_configuration_field_index position_in_array += 1 model_configuration_field_index = model_configuration_field_object[base_model_name_value] model_configuration_name_value = base_model_object_of_model_configuration_arrays[base_model_name_value][model_configuration_field_index] if base_model_name_value == "sdxl_turbo": negative_prompt_text = "" base_model_steps = base_model_steps_field_for_sdxl_turbo guidance_scale = 0 current_estimated_total_base_model_steps = base_model_steps current_estimated_total_refiner_steps = 0 current_estimated_total_upscaler_steps = upscaling_steps global current_actual_total_base_model_steps global current_actual_total_refiner_steps global current_actual_total_upscaler_steps current_actual_total_base_model_steps = 0 current_actual_total_refiner_steps = 0 current_actual_total_upscaler_steps = 0 scheduler_value = schedulers_array[scheduler_index] if not base_model_valid(base_model_name_value): error_function("Base model is not valid.") if not model_configuration_valid(base_model_name_value, model_configuration_name_value): error_function("Model configuration is not valid.") if not prompt_valid(prompt_text): error_function("Prompt is not valid.") if not negative_prompt_valid(negative_prompt_text): error_function("Negative prompt is not valid.") if not scheduler_valid(scheduler_value): error_function("Scheduler/sampler is not valid.") if not width_valid(image_width): error_function("Image width is not valid.") if not height_valid(image_height): error_function("Image height is not valid.") if base_model_name_value != "sdxl_turbo": if not guidance_scale_valid(guidance_scale): error_function("Guidance scale is not valid.") if not steps_valid(base_model_steps, base_model_name_value): error_function("Steps option is not valid.") if not seed_valid(actual_seed): error_function("Seed is not valid.") add_seed_into_pipe = numerical_bool(add_seed_into_pipe) use_torch_manual_seed_but_do_not_add_to_pipe_field = numerical_bool(use_torch_manual_seed_but_do_not_add_to_pipe_field) refining_selection_field_value = numerical_bool(refining_selection_field_value) refining_use_denoising_start_in_base_model_when_using_refiner_field_value = numerical_bool(refining_use_denoising_start_in_base_model_when_using_refiner_field_value) refining_base_model_output_to_refiner_is_in_latent_space_field_value = numerical_bool(refining_base_model_output_to_refiner_is_in_latent_space_field_value) use_upscaler = numerical_bool(upscaling_selection_field_value) use_refiner = 0 num_inference_steps_in_refiner = base_model_steps if refining_selection_field_value: use_refiner = 1 if not refiner_denoise_start_valid(refining_denoise_start_field_value): error_function("Refiner denoise start is not valid.") # if refining_steps_for_older_configuration_field_value > 0: # For older configrations. Doesn't reflect actual number of steps. # num_inference_steps_in_refiner = refining_steps_for_older_configuration_field_value if use_upscaler == 1: if not upscaling_steps_valid(upscaling_steps): error_function("Upscaling steps option is not valid.") if ( (last_model_configuration_name_selected_state_value == "") or (model_configuration_name_value != last_model_configuration_name_selected_state_value) ): if (last_model_configuration_name_selected_state_value != ""): if "pipe" in globals(): del pipe if device == "cuda": torch.cuda.empty_cache() if show_messages_in_command_prompt == 1: print ("Base model is loading."); progress( progress = 0, desc = "Base model is loading" ) ( pipe ) = construct_pipe( base_model_name_value, model_configuration_name_value ) last_model_configuration_name_selected_state_value = model_configuration_name_value else: pipe = stored_pipe_state ( scheduler_used ) = configure_scheduler( pipe, scheduler_value ) if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value): use_refiner = 0 use_upscaler = 0 if log_generation_times == 1: generation_start_time = time.time() # Only 77 tokens are allowed in the prompt. 2 are reserved, meaning it is # truncated to 75. This happens automatically, but we want to tell people # that tokenizer = pipe.tokenizer max_token_length_of_model = pipe.tokenizer.model_max_length - number_of_reserved_tokens token_length_of_prompt_text = len(tokenizer.tokenize(prompt_text)) token_length_of_negative_prompt_text = len(tokenizer.tokenize(negative_prompt_text)) prompt_truncated_field_group_update = gr.Group( visible = False ) prompt_truncated_field_update = gr.HTML( value = "" ) negative_prompt_truncated_field_group_update = gr.Group( visible = False ) negative_prompt_truncated_field_update = gr.HTML( value = "" ) prompt_text_not_used_substring = "" negative_prompt_text_not_used_substring = "" truncated_prompts = 0 partial_prompt_or_negative_prompt_length_too_long_message = "" if token_length_of_prompt_text > max_token_length_of_model: ( prompt_text, prompt_text_not_used_substring ) = truncate_prompt( pipe, prompt_text ) prompt_truncated_field_group_update = gr.Group( visible = True ) prompt_truncated_field_update = gr.Textbox( value = prompt_text_not_used_substring ) truncated_prompts += 1 partial_prompt_or_negative_prompt_length_too_long_message += "prompt" if token_length_of_negative_prompt_text > max_token_length_of_model: ( negative_prompt_text, negative_prompt_text_not_used_substring ) = truncate_prompt( pipe, negative_prompt_text ) negative_prompt_truncated_field_group_update = gr.Group( visible = True ) negative_prompt_truncated_field_update = gr.Textbox( value = negative_prompt_text_not_used_substring ) truncated_prompts += 1 if truncated_prompts == 2: partial_prompt_or_negative_prompt_length_too_long_message += " and " partial_prompt_or_negative_prompt_length_too_long_message += "negative prompt" if len(partial_prompt_or_negative_prompt_length_too_long_message) > 0: partial_prompt_or_negative_prompt_length_too_long_message += " was" if truncated_prompts == 2: partial_prompt_or_negative_prompt_length_too_long_message += " were" prompt_or_negative_prompt_length_too_long_message = "Note: Part of your " + partial_prompt_or_negative_prompt_length_too_long_message + " truncated automatically because it was too long." show_message(prompt_or_negative_prompt_length_too_long_message) if add_seed_into_pipe == 1: generator = torch.manual_seed(actual_seed) else: if use_torch_manual_seed_but_do_not_add_to_pipe_field == 1: torch.manual_seed(actual_seed) generator = None denoising_end_applicable = 0 if base_model_name_value not in base_models_not_supporting_denoising_end_for_base_model_object: denoising_end_applicable = 1 denoising_end_in_base_model_to_use = None output_type_in_base_model_to_use = "pil" if use_refiner == 1: if ( (refining_use_denoising_start_in_base_model_when_using_refiner_field_value == 1) and (denoising_end_applicable == 1) ): denoising_end_in_base_model_to_use = refining_denoise_start_field_value current_estimated_total_base_model_steps = rounded_number(base_model_steps * refining_denoise_start_field_value) if current_estimated_total_base_model_steps < 1: current_estimated_total_base_model_steps = 1 if refining_base_model_output_to_refiner_is_in_latent_space_field_value == 1: output_type_in_base_model_to_use = "latent" current_estimated_total_refiner_steps = rounded_number(base_model_steps - (base_model_steps * refining_denoise_start_field_value)) if current_estimated_total_refiner_steps < 1: current_estimated_total_refiner_steps = 1 upscaled_image_width = 0 upscaled_image_height = 0 if use_upscaler == 1: upscaled_image_width = int(image_width * 2) upscaled_image_height = int(image_height * 2) current_base_model_generation_start_time = 0 global saved_final_base_model_pil_image_if_using_refiner saved_final_base_model_pil_image_if_using_refiner = None global upscaled_image_canceled global upscaled_image_canceled_latents upscaled_image_canceled = 0 # upscaled_image_canceled_latents = None def callback_function_for_base_model_progress( callback_pipe, callback_step_index, callback_timestep, callback_kwargs ): callback_step_number = (int(callback_step_index) + 1) global current_actual_total_base_model_steps current_actual_total_base_model_steps += 1 global current_base_model_generation_start_time if callback_step_number == 1: current_base_model_generation_start_time = time.time() seconds_per_step = 0 if callback_step_number > 1: seconds_per_step = ((time.time() - current_base_model_generation_start_time) / int(callback_step_index)) nice_time_per_step = nice_elapsed_time(seconds_per_step) base_model_progress_text = nice_time_per_step + " per step" else: base_model_progress_text = "Base model processing started" cancel_process = image_processing_is_canceled(user_id_state_value, image_generation_id_state_value) if cancel_process: pipe._interrupt = True if ( use_image_preview and ( ((int(callback_step_index) % image_preview_step_interval) == 0) or (seconds_per_step >= image_preview_seconds_interval) ) and (callback_step_number < current_estimated_total_base_model_steps) ): latents = callback_kwargs["latents"] temporary_extra = str(user_id_state_value) + "_base_model_" + str(callback_step_number) model_to_use = base_model_name_value is_final_image = 0 pil_image = create_image_from_latents( model_to_use, pipe, latents, generator, is_final_image ) create_preview_image( model_to_use, user_id_state_value, pipe, latents, generator, temporary_extra ) if ( cancel_process or ( (base_model_steps == callback_step_number) and (use_refiner == 1) ) ): global saved_final_base_model_pil_image_if_using_refiner latents = callback_kwargs["latents"] is_final_image = 1 saved_final_base_model_pil_image_if_using_refiner = create_image_from_latents( base_model_name_value, pipe, latents, generator, is_final_image ) if ( (show_image_creation_progress_log == 1) and (callback_step_number <= current_estimated_total_base_model_steps) ): progress( progress = ( callback_step_number, current_estimated_total_base_model_steps ), desc = base_model_progress_text, unit = "base model steps" ) return {} callback_to_do_for_base_model_progress = callback_function_for_base_model_progress if ( (show_image_creation_progress_log == 1) or enable_image_generation_cancellation or use_image_preview ): current_refiner_generation_start_time = 0 def callback_function_for_refiner_progress( callback_pipe, callback_step_index, callback_timestep, callback_kwargs ): callback_step_number = (int(callback_step_index) + 1) global current_actual_total_refiner_steps current_actual_total_refiner_steps += 1 global current_refiner_generation_start_time if callback_step_number == 1: current_refiner_generation_start_time = time.time() seconds_per_step = 0 if callback_step_number > 1: seconds_per_step = ((time.time() - current_refiner_generation_start_time) / int(callback_step_index)) nice_time_per_step = nice_elapsed_time(seconds_per_step) refiner_progress_text = nice_time_per_step + " per step" else: refiner_progress_text = "Refiner processing started" if ( use_image_preview and ( ((int(callback_step_index) % image_preview_step_interval) == 0) or (seconds_per_step >= image_preview_seconds_interval) ) and (callback_step_number < current_estimated_total_refiner_steps) ): latents = callback_kwargs["latents"] temporary_extra = str(user_id_state_value) + "_refiner_" + str(callback_step_number) model_to_use = base_model_name_value create_preview_image( model_to_use, user_id_state_value, pipe, latents, generator, temporary_extra ) if ( (show_image_creation_progress_log == 1) and (callback_step_number <= current_estimated_total_refiner_steps) ): progress( progress = ( callback_step_number, current_estimated_total_refiner_steps ), desc = refiner_progress_text, unit = "est. refiner steps" ) if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value): refiner._interrupt = True return {} callback_to_do_for_refiner_progress = callback_function_for_refiner_progress current_upscaler_generation_start_time = 0 def callback_function_for_upscaler_progress( callback_step_index, callback_timestep, callback_latents ): callback_step_number = (int(callback_step_index) + 1) global current_actual_total_upscaler_steps current_actual_total_upscaler_steps += 1 global current_upscaler_generation_start_time if callback_step_number == 1: current_upscaler_generation_start_time = time.time() seconds_per_step = 0 if callback_step_number > 1: seconds_per_step = ((time.time() - current_upscaler_generation_start_time) / int(callback_step_index)) nice_time_per_step = nice_elapsed_time(seconds_per_step) upscaler_progress_text = nice_time_per_step + " per step" else: upscaler_progress_text = "Upscaler processing started" if ( use_image_preview and ( ((int(callback_step_index) % image_preview_step_interval) == 0) or (seconds_per_step >= image_preview_seconds_interval) ) and (callback_step_number < current_estimated_total_upscaler_steps) ): latents = callback_latents temporary_extra = str(user_id_state_value) + "_upscale_" + str(callback_step_index) model_to_use = "upscaler" create_preview_image( model_to_use, user_id_state_value, pipe, latents, generator, temporary_extra ) if ( (show_image_creation_progress_log == 1) and (callback_step_number <= current_estimated_total_upscaler_steps) ): progress( progress = ( callback_step_number, current_estimated_total_upscaler_steps ), desc = upscaler_progress_text, unit = "upscaler steps" ) if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value): # This doesn't work here. We raise an excpetion to cancel # instead. # upscaler._interrupt = True global upscaled_image_canceled global upscaled_image_canceled_latents upscaled_image_canceled = 1 latents = callback_latents upscaled_image_canceled_latents = latents raise Exception("end_at_this_step") return {} callback_to_do_for_upscaler_progress = callback_function_for_upscaler_progress else: callback_to_do_for_base_model_progress = None callback_to_do_for_refiner_progress = None callback_to_do_for_upscaler_progress = None if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value): if show_messages_in_command_prompt == 1: print ("Image processing was canceled before any part of the image could be created."); remove_from_cancel_object(user_id_state_value, image_generation_id_state_value) output_text_field_update = gr.Textbox() return { output_text_field: output_text_field_update } # output_image_field_update = gr.Image() # output_image_gallery_field_update = gr.Gallery() # output_text_field_update = gr.Textbox() # prompt_truncated_field_group_update = gr.Group() # prompt_truncated_field_update = gr.Textbox() # negative_prompt_truncated_field_group_update = gr.Group() # negative_prompt_truncated_field_update = gr.Textbox() # error_text_field_update = gr.Textbox() # return ( # output_image_field_update, # output_image_gallery_field_update, # output_text_field_update, # prompt_truncated_field_group_update, # prompt_truncated_field_update, # negative_prompt_truncated_field_group_update, # negative_prompt_truncated_field_update, # last_model_configuration_name_selected_state_value, # last_refiner_name_selected_state_value, # last_upscaler_name_selected_state_value, # pipe, # stored_refiner_state, # stored_upscaler_state, # error_text_field_update # ) task_info_for_progress = "Initial image creation has begun" if use_refiner == 1: if use_upscaler == 1: task_info_for_command_prompt = "Will create initial image, then refine and then upscale.\nInitial image steps..." else: task_info_for_command_prompt = "Will create initial image and then refine.\nInitial image steps..." else: if use_upscaler == 1: task_info_for_command_prompt = "Will create initial image and then upscale.\nInitial image steps..." else: task_info_for_command_prompt = "Will create image (no refining or upscaling).\nImage steps..." task_info_for_progress = "Image creation has begun" if show_messages_in_command_prompt == 1: print (task_info_for_command_prompt); if show_image_creation_progress_log == 1: progress( progress = 0, desc = task_info_for_progress ) base_image = pipe( prompt = prompt_text, negative_prompt = negative_prompt_text, width = image_width, height = image_height, num_inference_steps = base_model_steps, guidance_scale = guidance_scale, num_images_per_prompt = 1, generator = generator, denoising_end = denoising_end_in_base_model_to_use, output_type = output_type_in_base_model_to_use, callback_on_step_end = callback_to_do_for_base_model_progress ) have_upscaled_image = 0 have_refined_image = 0 base_image_for_next_step = base_image.images refined_image_for_next_step = None upscaled_image_for_next_step = None error_count = 0 error_array = [] if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value): use_refiner = 0 use_upscaler = 0 refiner = {} upscaler = {} refiner_error = 0 upscaler_error = 0 if ( (use_refiner == 1) or (use_upscaler == 1) ): image_for_next_step = base_image_for_next_step if use_refiner == 1: try: if (last_refiner_name_selected_state_value == ""): if show_messages_in_command_prompt == 1: print ("Refiner is loading."); progress( progress = 0, desc = "Refiner is loading" ) refiner = construct_refiner() last_refiner_name_selected_state_value = "refiner" else: refiner = stored_refiner_state except BaseException as error_message: use_refiner = 0 refiner_error = 1 error_count += 1 error_array.extend([ "Error " + str(error_count) + ": An error occurred while trying to load the refiner:\n" + str(error_message) ]) if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value): use_refiner = 0 use_upscaler = 0 if use_refiner == 1: if show_messages_in_command_prompt == 1: print ("Refiner steps..."); if show_image_creation_progress_log == 1: progress( progress = 0, desc = "Refining is beginning" ) try: refined_image = refiner( prompt = prompt_text, negative_prompt = negative_prompt_text, image = base_image_for_next_step, num_inference_steps = num_inference_steps_in_refiner, denoising_start = refining_denoise_start_field_value, output_type = "pil", generator = generator, callback_on_step_end = callback_to_do_for_refiner_progress ) except BaseException as error_message: # User chose to refine image, but something went wrong. # We won't use the refiner. Since we will be using the # base image, where the output could be latents, we'll # need to handle that later. use_refiner = 0 use_upscaler = 0 refiner_error = 1 error_count += 1 error_array.extend([ "Error " + str(error_count) + ": An error occurred while refining:\n" + str(error_message) ]) if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value): use_upscaler = 0 if use_refiner == 1: # User chose to refine image and it succeded. have_refined_image = 1 refined_image_for_next_step = refined_image.images[0] image_for_next_step = refined_image_for_next_step if use_upscaler == 1: try: if (last_upscaler_name_selected_state_value == ""): if show_messages_in_command_prompt == 1: print ("Upscaler is loading."); progress( progress = 0, desc = "Upscaler is loading" ) upscaler = construct_upscaler() last_upscaler_name_selected_state_value = "upscaler" else: upscaler = stored_upscaler_state except BaseException as error_message: use_upscaler = 0 upscaler_error = 1 error_count += 1 error_array.extend([ "Error " + str(error_count) + ": An error occurred while trying to load the upscaler:\n" + str(error_message) ]) if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value): use_upscaler = 0 if use_upscaler == 1: if show_messages_in_command_prompt == 1: print ("Upscaler steps..."); if show_image_creation_progress_log == 1: progress( progress = 0, desc = "Upscaling is beginning" ) try: upscaled_image = upscaler( prompt = prompt_text, negative_prompt = negative_prompt_text, image = image_for_next_step, num_inference_steps = upscaling_steps, guidance_scale = 0, generator = generator, callback = callback_to_do_for_upscaler_progress, callback_steps = 1 ) except BaseException as error_message: use_upscaler = 0 if str(error_message) != "end_at_this_step": # User chose to upscale image, but something went # wrong. We won't use the upscaler. # # If "error_message" is "end_at_this_step", then the # user chose to cancel. We have to handle canceling # when upscaling differently. We don't want to treat # this as an actual error, so we don't do the below. upscaler_error = 1 error_count += 1 error_array.extend([ "Error " + str(error_count) + ":\nAn error occurred while upscaling:\n" + str(error_message) ]) if use_upscaler == 1: # User chose to upscale image and it succeded. have_upscaled_image = 1 upscaled_image_for_next_step = upscaled_image.images[0] if upscaled_image_canceled == 1: latents = upscaled_image_canceled_latents temporary_extra = str(user_id_state_value) + "_upscale_canceled" model_to_use = "upscaler" is_final_image = 1 pil_image = create_image_from_latents( model_to_use, pipe, latents, generator, is_final_image ) for key, value in enumerate(pil_image): image_to_return = value elif have_upscaled_image == 1: image_to_return = upscaled_image_for_next_step elif have_refined_image == 1: image_to_return = refined_image_for_next_step else: if output_type_in_base_model_to_use == "latent": # Image is in latent space for key, value in enumerate(saved_final_base_model_pil_image_if_using_refiner): image_to_return = value else: image_to_return = base_image_for_next_step[0] image_has_been_canceled = 0 if image_processing_is_canceled(user_id_state_value, image_generation_id_state_value): image_has_been_canceled = 1 if device == "cuda": torch.cuda.empty_cache() which_image = "" image_generation_information = create_image_generation_information( base_model_name_value, model_configuration_name_value, scheduler_used, scheduler_value, prompt_text, prompt_text_not_used_substring, negative_prompt_text, negative_prompt_text_not_used_substring, image_width, image_height, actual_seed, add_seed_into_pipe, guidance_scale, base_model_steps, display_xformers_usage_in_prompt_info, display_default_attn_processor_usage_in_prompt_info, display_diffusers_version_in_prompt_info, use_refiner, refiner_error, refining_denoise_start_field_value, denoising_end_applicable, refining_use_denoising_start_in_base_model_when_using_refiner_field_value, refining_base_model_output_to_refiner_is_in_latent_space_field_value, refining_steps_for_older_configuration_field_value, use_upscaler, upscaler_error, upscaling_steps, upscaled_image_width, upscaled_image_height, current_actual_total_base_model_steps, current_actual_total_refiner_steps, current_actual_total_upscaler_steps, generation_start_time, image_has_been_canceled, which_image ) # have_refined_image, # have_upscaled_image, #Image before refining #Image before upscaling output_text_field_update = gr.Textbox( value = image_generation_information, lines = 12 ) if add_generation_information_to_image == 1: # Add generation info to image info_to_save_in_image = "\n-----------\nImage generation information:\n" + image_generation_information + "\n-----------\n" image_to_return.info = {"parameters": info_to_save_in_image} # save_base_image_when_using_refiner_field_value, # save_refined_image_when_using_upscaler_field_value, save_canceled_images = 0 if ( (auto_save_imagery == 1) and ( (image_has_been_canceled == 0) or (save_canceled_images == 1) ) ): if not os.path.exists(saved_images_dir): os.makedirs(saved_images_dir) yy_mm_dd_date_stamp = datetime.today().strftime('%Y-%m-%d') saved_images_date_dir = saved_images_dir + "/" + yy_mm_dd_date_stamp + "/" if not os.path.exists(saved_images_date_dir): os.makedirs(saved_images_date_dir) image_count = 1 file_name_without_extension = yy_mm_dd_date_stamp + "-" + ('%04d' % image_count) saved_image_path_and_file = saved_images_date_dir + file_name_without_extension + ".png" while os.path.exists(saved_image_path_and_file): file_name_without_extension = yy_mm_dd_date_stamp + "-" + ('%04d' % image_count) saved_image_path_and_file = saved_images_date_dir + file_name_without_extension + ".png" image_count += 1 final_image_file_path_and_file = saved_images_date_dir + file_name_without_extension + ".png" save_image_file( final_image_file_path_and_file, image_to_return, add_generation_information_to_image, info_to_save_in_image, ) final_image_text_file_path_and_file = saved_images_date_dir + file_name_without_extension + ".txt" prompt_info_file_handle = open(final_image_text_file_path_and_file, "w") prompt_info_file_handle.writelines(image_generation_information) prompt_info_file_handle.close() output_image_field_update = gr.Image( value = image_to_return ) image_gallery_array_state_value.insert(0, image_to_return) prompt_information_array_state_value.insert(0, image_generation_information) output_image_gallery_field_update = gr.Gallery( value = image_gallery_array_state_value, selected_index = 0 ) image_gallery_array_state_update = image_gallery_array_state_value prompt_information_array_state_update = prompt_information_array_state_value error_text_field_update = gr.Textbox( visible = False ) if (len(error_array) > 0): error_information = "\n\n".join(error_array) error_text_field_update = gr.Textbox( value = error_information, visible = True ) if show_messages_in_command_prompt == 1: print ("Image created.") last_model_configuration_name_selected_state_update = last_model_configuration_name_selected_state_value last_refiner_name_selected_state_update = last_refiner_name_selected_state_value last_upscaler_name_selected_state_update = last_upscaler_name_selected_state_value current_preview_image = None current_preview_image_user_id = 0 if image_has_been_canceled == 1: remove_from_cancel_object(user_id_state_value, image_generation_id_state_value) return ( output_image_field_update, output_image_gallery_field_update, output_text_field_update, prompt_truncated_field_group_update, prompt_truncated_field_update, negative_prompt_truncated_field_group_update, negative_prompt_truncated_field_update, last_model_configuration_name_selected_state_update, last_refiner_name_selected_state_update, last_upscaler_name_selected_state_update, pipe, refiner, upscaler, error_text_field_update ) ##################### # # Cancel Image Function # # Image generation has not yet started and we need to cancel it. # ##################### def cancel_image_function( user_id_state, image_generation_id_state ): user_id_state_value = user_id_state.value image_generation_id_state_value = image_generation_id_state.value global image_generation_ids_object global cancel_image_generation_times_object if (user_id_state_value not in cancel_image_generation_ids_object): cancel_image_generation_ids_object[user_id_state_value] = [] cancel_image_generation_ids_object[user_id_state_value].append(image_generation_id_state_value) cancel_image_generation_times_object[image_generation_id_state_value] = time.time() cancel_image_button_update = gr.Button( value = cancel_image_button_in_progress_text, interactive = False ) cancel_image_message_field_row_update = gr.Row( visible = True ) cancel_image_object = {} if image_generation_id_state_value == current_image_generation_id_in_progress: # If they cancel the image that is currently being generated, we will # stop it after the next step is complete and return the image to # them. They will not need to reload the page again to create another # image. cancel_image_message_field_html = canceled_image_in_process_of_being_generated show_message("Your image generation is being canceled. After the current step in the process is complete, the cancellation will be complete.") generate_image_button_row_update = gr.Row( visible = False ) cancel_image_object.update({ generate_image_button_row: generate_image_button_row_update, cancel_image_message_field_row: cancel_image_message_field_row_update }) else: # If they cancel there place in the queue, we have to make them # reload the page if they want to generate another image. cancel_image_message_field_html = canceled_image_in_queue_message generate_image_button_update = gr.Button( value = generate_image_button_normal_text, interactive = False ) output_text_field_update = gr.Textbox( visible = True ) output_image_preview_field_row_update = gr.Row( visible = False ) cancel_image_button_row_update = gr.Row( visible = False ) cancel_image_object.update({ generate_image_button: generate_image_button_update, output_text_field: output_text_field_update, output_image_preview_field_row: output_image_preview_field_row_update, cancel_image_button_row: cancel_image_button_row_update, cancel_image_message_field_row: cancel_image_message_field_row_update }) cancel_image_message_field_update = gr.Button( value = cancel_image_message_field_html ) cancel_image_object.update({ cancel_image_button: cancel_image_button_update, cancel_image_message_field_row: cancel_image_message_field_row_update, cancel_image_message_field: cancel_image_message_field_update }) return cancel_image_object ##################### # # Cancel Image Processing # # When running on Windows, this is an attempt at closing the command # prompt from the web display. It's really not worth having this. You can # just close the prompt. I would like a nice way to cancel image # creation, but couldn't figure that out. # ##################### def close_command_prompt(): # I simply don't know how to stop the image generation without closing # the command prompt. Doing that requires the code below twice for some # reason. # # Method: # https://stackoverflow.com/questions/67146623/how-to-close-the-command-prompt-from-python-script-directly gr.Warning("The command prompt window has been closed. Any image generation in progress has been stopped. To generate any other images, you will need to launch the command prompt again.") os.system('title kill_window') os.system(f'taskkill /f /fi "WINDOWTITLE eq kill_window"') os.system(f'taskkill /f /fi "WINDOWTITLE eq kill_window"') ##################### # # Download Data From HuggingFace # # This will download a lot of data at once rather than waiting until you # use each model. # ##################### def download_data_from_huggingface( download_data_option ): if ( (script_being_run_on_hugging_face == 0) and ("HF_HUB_OFFLINE" in os.environ) and (int(os.environ["HF_HUB_OFFLINE"]) == 0) ): data_to_get_partial_message = "the default model configuration defined in \"base_model_model_configuration_defaults_object\" for that model will be downloaded. It accesses data that is linked in \"model_configuration_links_object\"." if download_data_option == "2": data_to_get_partial_message = "all model data, for each model configuration, will be downloaded. This is defined in \"base_model_object_of_model_configuration_arrays\" and accesses data that is linked in \"model_configuration_links_object\"." download_data_message = "For each model in the model dropdown (\"base_model_array\"), " + data_to_get_partial_message + " That could easily be dozens of gigabytes of data or more that is about to be downloaded. If you want to stop the download, close the command prompt." print (download_data_message) data_links_downloaded_object = {} for this_base_model in base_model_array: base_model_name_value = this_base_model if download_data_option == "1": default_model_configuration_for_this_base_model = base_model_model_configuration_defaults_object[this_base_model] print ("Downloading/loading \"" + this_base_model + "\" model data for \"" + default_model_configuration_for_this_base_model + "\"...") model_configuration_name_value = default_model_configuration_for_this_base_model construct_pipe ( base_model_name_value, model_configuration_name_value ) else: for this_model_configuration in base_model_object_of_model_configuration_arrays[this_base_model]: if ( (this_model_configuration in model_configuration_names_object) and (this_model_configuration in model_configuration_links_object) ): model_configuration_name_value = this_model_configuration model_configuration_link_value = model_configuration_links_object[this_model_configuration] if model_configuration_link_value not in data_links_downloaded_object: print ("Downloading/loading \"" + this_base_model + "\" model data from \"" + model_configuration_link_value + "\"...") construct_pipe ( base_model_name_value, model_configuration_name_value ) data_links_downloaded_object[model_configuration_link_value] = 1 print ("Downloading/loading refiner data...") construct_refiner() print ("Downloading/loading upscaler data...") construct_upscaler() print ("The data has been downloaded.") else: error_function("In order to download model data, \"HF_HUB_OFFLINE\" must be set to \"0\" in the Windows .bat file that launched this script.") ##################### # # Get Query Params # # Get variables from the url of the page and update the display to # reflect them. # ##################### def get_query_params( request: gr.Request ): raw_url_params = str(request.query_params) import urllib.parse unprocessed_url_object = urllib.parse.parse_qs(raw_url_params) url_object = {} for url_param_key in unprocessed_url_object: url_param_value = unprocessed_url_object[url_param_key][0] if len(url_param_value) > 0: url_param_key = str(url_param_key) url_param_key_lowercase = url_param_key.lower() url_object[url_param_key_lowercase] = str(unprocessed_url_object[url_param_key][0]).lower() field_object = {} user_id_number = int(random.randrange(0, 1000000000)) user_id_state_update = gr.State(user_id_number) field_object.update({user_id_state: user_id_state_update}) base_model_name_value = default_base_model base_model_field_key_in_url = "model" if base_model_field_key_in_url in url_object: base_model_field_in_url = url_object[base_model_field_key_in_url].lower() if base_model_valid(base_model_field_in_url): base_model_name_value = base_model_field_in_url base_model_nicely_named_value = base_model_names_object[base_model_name_value] field_object.update({base_model_field: base_model_nicely_named_value}) download_data_key_in_url = "download_data" if download_data_key_in_url in url_object: download_data_option_in_url = str(url_object[download_data_key_in_url]) if ( (download_data_option_in_url == "1") or (download_data_option_in_url == "2") ): download_data_from_huggingface(download_data_option_in_url) model_configuration_key_in_url = "model_config" model_configuration_in_url = "" if model_configuration_key_in_url in url_object: model_configuration_in_url = url_object[model_configuration_key_in_url].lower() for this_base_model in base_model_array: if base_model_name_value == this_base_model: model_configuration_name_value = base_model_model_configuration_defaults_object[this_base_model] if len(model_configuration_in_url) > 0: for this_model_configuration in base_model_object_of_model_configuration_arrays[this_base_model]: if model_configuration_valid(base_model_name_value, model_configuration_in_url): model_configuration_name_value = this_model_configuration field_object.update({initial_model_configuration_name_selected_state: model_configuration_in_url}) prompt_field_key_in_url = "prompt" if prompt_field_key_in_url in url_object: prompt_field_in_url = url_object[prompt_field_key_in_url].lower() if prompt_valid(prompt_field_in_url): field_object.update({prompt_field: prompt_field_in_url}) negative_prompt_field_key_in_url = "neg_prompt" if negative_prompt_field_key_in_url in url_object: negative_prompt_field_in_url = url_object[negative_prompt_field_key_in_url].lower() if prompt_valid(negative_prompt_valid): field_object.update({negative_prompt_field: negative_prompt_field_in_url}) scheduler_field_key_in_url = "scheduler" if scheduler_field_key_in_url in url_object: scheduler_field_in_url = url_object[scheduler_field_key_in_url].lower() if scheduler_valid(scheduler_field_in_url): scheduler_name_value = scheduler_field_in_url scheduler_nicely_named_value = scheduler_long_names_object[scheduler_name_value] field_object.update({scheduler_field: scheduler_nicely_named_value}) image_width_field_key_in_url = "width" if image_width_field_key_in_url in url_object: image_width_field_in_url = str(url_object[image_width_field_key_in_url]) if width_valid(image_width_field_in_url): field_object.update({image_width_field: image_width_field_in_url}) image_height_field_key_in_url = "height" if image_height_field_key_in_url in url_object: image_height_field_in_url = str(url_object[image_height_field_key_in_url]) if height_valid(image_height_field_in_url): field_object.update({image_height_field: image_height_field_in_url}) guidance_scale_field_key_in_url = "guidance" if guidance_scale_field_key_in_url in url_object: guidance_scale_field_in_url = str(url_object[guidance_scale_field_key_in_url]) if guidance_scale_valid(guidance_scale_field_in_url): field_object.update({guidance_scale_field: guidance_scale_field_in_url}) steps_key_in_url = "steps" if steps_key_in_url in url_object: steps_in_url = str(url_object[steps_key_in_url]) if steps_valid(steps_in_url, base_model_name_value): if base_model_name_value == "sdxl_turbo": field_object.update({base_model_steps_field_for_sdxl_turbo_field: steps_in_url}) else: field_object.update({base_model_steps_field: steps_in_url}) seed_field_key_in_url = "seed" if seed_field_key_in_url in url_object: seed_field_in_url = url_object[seed_field_key_in_url] if seed_valid(seed_field_in_url): field_object.update({seed_field: seed_field_in_url}) add_seed_key_in_url = "add_seed" add_seed_to_generation = None if add_seed_key_in_url in url_object: add_seed_in_url = url_object[add_seed_key_in_url].lower() add_seed_to_generation = True if ( (add_seed_in_url == "0") or (add_seed_in_url == "n") or (add_seed_in_url == "no") or (add_seed_in_url == "false") ): add_seed_to_generation = False field_object.update({add_seed_into_pipe_field: add_seed_to_generation}) use_torch_manual_seed_but_not_in_generator_key_in_url = "use_torch_manual_seed_but_not_in_generator" if use_torch_manual_seed_but_not_in_generator_key_in_url in url_object: use_torch_manual_seed_but_not_in_generator_in_url = url_object[use_torch_manual_seed_but_not_in_generator_key_in_url].lower() use_torch_manual_seed_but_not_in_generator_to_generation = False if ( (use_torch_manual_seed_but_not_in_generator_in_url == "1") or (use_torch_manual_seed_but_not_in_generator_in_url == "y") or (use_torch_manual_seed_but_not_in_generator_in_url == "yes") or (use_torch_manual_seed_but_not_in_generator_in_url == "true") ): use_torch_manual_seed_but_not_in_generator_to_generation = True if ( add_seed_to_generation and add_seed_to_generation == True ): use_torch_manual_seed_but_not_in_generator_to_generation = False field_object.update({use_torch_manual_seed_but_do_not_add_to_pipe_field: use_torch_manual_seed_but_not_in_generator_to_generation}) refiner_key_in_url = "refiner" if refiner_key_in_url in url_object: refiner_in_url = url_object[refiner_key_in_url].lower() refiner_in_url_formatted = "No" if ( (refiner_in_url == "1") or (refiner_in_url == "y") or (refiner_in_url == "yes") or (refiner_in_url == "true") ): refiner_in_url_formatted = "Yes" field_object.update({refining_selection_field: refiner_in_url_formatted}) refiner_denoise_start_key_in_url = "denoise_start" if refiner_denoise_start_key_in_url in url_object: refiner_denoise_start_in_url = str(url_object[refiner_denoise_start_key_in_url]) if refiner_denoise_start_valid(refiner_denoise_start_in_url): field_object.update({refining_denoise_start_field: refiner_denoise_start_in_url}) refining_steps_key_in_url = "refiner_steps" if refining_steps_key_in_url in url_object: refining_steps_in_url = str(url_object[refining_steps_key_in_url]) if refining_steps_valid(refining_steps_in_url): field_object.update({refining_steps_for_older_configuration_field: refining_steps_in_url}) use_denoising_start_in_base_model_when_using_refiner_key_in_url = "use_denoise_end" if use_denoising_start_in_base_model_when_using_refiner_key_in_url in url_object: use_denoising_start_in_base_model_when_using_refiner_in_url = url_object[use_denoising_start_in_base_model_when_using_refiner_key_in_url].lower() use_denoising_start_in_base_model_when_using_refiner_bool = True if ( (use_denoising_start_in_base_model_when_using_refiner_in_url == "0") or (use_denoising_start_in_base_model_when_using_refiner_in_url == "n") or (use_denoising_start_in_base_model_when_using_refiner_in_url == "no") or (use_denoising_start_in_base_model_when_using_refiner_in_url == "false") ): use_denoising_start_in_base_model_when_using_refiner_bool = False field_object.update({refining_use_denoising_start_in_base_model_when_using_refiner_field: use_denoising_start_in_base_model_when_using_refiner_bool}) base_model_output_to_refiner_is_in_latent_space_key_in_url = "latent_space_before_refiner" if base_model_output_to_refiner_is_in_latent_space_key_in_url in url_object: base_model_output_to_refiner_is_in_latent_space_in_url = url_object[base_model_output_to_refiner_is_in_latent_space_key_in_url].lower() base_model_output_to_refiner_is_in_latent_space_bool = True if ( (base_model_output_to_refiner_is_in_latent_space_in_url == "0") or (base_model_output_to_refiner_is_in_latent_space_in_url == "n") or (base_model_output_to_refiner_is_in_latent_space_in_url == "no") or (base_model_output_to_refiner_is_in_latent_space_in_url == "false") ): base_model_output_to_refiner_is_in_latent_space_bool = False field_object.update({refining_base_model_output_to_refiner_is_in_latent_space_field: base_model_output_to_refiner_is_in_latent_space_bool}) upscaler_key_in_url = "upscaler" if upscaler_key_in_url in url_object: upscaler_in_url = url_object[upscaler_key_in_url].lower() upscaler_in_url_formatted = "No" if ( (upscaler_in_url == "1") or (upscaler_in_url == "y") or (upscaler_in_url == "yes") or (upscaler_in_url == "true") ): upscaler_in_url_formatted = "Yes" field_object.update({upscaling_selection_field: upscaler_in_url_formatted}) upscaling_steps_key_in_url = "upscaler_steps" if upscaling_steps_key_in_url in url_object: upscaling_steps_in_url = str(url_object[upscaling_steps_key_in_url]) if upscaling_steps_valid(upscaling_steps_in_url): field_object.update({upscaling_num_inference_steps_field: upscaling_steps_in_url}) generate_image_button_update = gr.Button( interactive = True ) field_object.update({generate_image_button: generate_image_button_update}) return field_object ##################### # # Set Base Model and Model Configuration from query_params # # We need to handle this separate because the model configuration needs # to return those field components in order. They are not named as they # are dynamic. # ##################### def set_base_model_and_model_configuration_from_query_params( base_model_field_index, initial_model_configuration_name_selected_state_value, *model_configuration_dropdown_fields_array ): base_model_name_value = base_model_array[base_model_field_index] model_configuration_name_value_for_selected_base_model = initial_model_configuration_name_selected_state_value model_configuration_dropdown_fields_array = [] for this_base_model in base_model_array: model_configuration_name_default_value_for_this_base_model = base_model_model_configuration_defaults_object[this_base_model] for this_model_configuration in base_model_object_of_model_configuration_arrays[this_base_model]: if ( (base_model_name_value == this_base_model) and (model_configuration_name_value_for_selected_base_model == this_model_configuration) ): model_configuration_name_default_value_for_this_base_model = model_configuration_name_value_for_selected_base_model this_configuration_field_default_value = model_configuration_names_object[model_configuration_name_default_value_for_this_base_model] this_configuration_field = gr.Dropdown( value = this_configuration_field_default_value ) model_configuration_dropdown_fields_array.append(this_configuration_field) base_model_and_model_configuration_return_outputs = [] for this_model_configuration_dropdown_field in model_configuration_dropdown_fields_array: base_model_and_model_configuration_return_outputs.append( this_model_configuration_dropdown_field ) return base_model_and_model_configuration_return_outputs ############################################################################### ############################################################################### # # # # # # # Create Web Display # # # # # # ############################################################################### ############################################################################### css_to_use = """ /* Hide border on image preview */ .generating {border: none !important;} /* Hide footer */ footer { display: none !important; } /* Dropdowns */ .sp_dropdown ul { min-width: auto; max-width: fit-content !important; max-height: 250px !important; } /* Checkboxes */ .sp_checkbox label { align-items: start; } .sp_checkbox label input { margin-top: 3px; } /* Scrollbar on prompt information */ .textbox_vertical_scroll textarea { resize: none; overflow-y: auto !important; } .textbox_vertical_scroll textarea::-webkit-scrollbar { width: 15px; } .textbox_vertical_scroll textarea::-webkit-scrollbar-track { background-color: rgb(245, 245, 245); } .textbox_vertical_scroll textarea::-webkit-scrollbar-track:hover { background-color: rgb(242, 242, 242); } .textbox_vertical_scroll textarea::-webkit-scrollbar-thumb { background-color: rgb(214, 214, 214); width: 100%; height: 60px; max-height: 80%; border: 1px solid rgb(224, 224, 224); } .textbox_vertical_scroll textarea::-webkit-scrollbar-thumb:hover { background-color: rgb(184, 184, 184); } .image_scaling button.image-button img { object-fit: scale-down; } /* Size of image for image preview */ #image_preview_id div.image-container { width: 100%; } #image_preview_id img { max-height: 300px; max-width: 300px; # max-height: 100%; # max-width: 100%; object-fit: scale-down; margin: 0 auto; } """ with gr.Blocks( title = "Spaghetti AI", css = css_to_use, theme = gr.themes.Default( spacing_size = "sm", # sm, md, lg radius_size = "sm" # none, sm, md, lg ), analytics_enabled = False ) as sd_interface: # Variables to store for user session image_gallery_array_state = gr.State([]) prompt_information_array_state = gr.State([]) initial_model_configuration_name_selected_state = gr.State("") last_model_configuration_name_selected_state = gr.State("") last_refiner_name_selected_state = gr.State("") last_upscaler_name_selected_state = gr.State("") stored_pipe_state = gr.State({}) stored_refiner_state = gr.State({}) stored_upscaler_state = gr.State({}) user_id_state = gr.State(0) image_generation_id_state = gr.State("") canceled_images_array_state = gr.State([]) gr.Markdown(opening_html) with gr.Row(): with gr.Column( scale = 3, min_width = 200 ): with gr.Row( visible = True ) as generate_image_button_row: generate_image_button = gr.Button( value = generate_image_button_normal_text, variant = "primary", interactive = False ) with gr.Row( visible = False ) as cancel_image_button_row: cancel_image_button = gr.Button( value = cancel_image_button_text, variant = "stop", interactive = False ) with gr.Row( visible = False ) as cancel_image_message_field_row: cancel_image_message_field = gr.HTML( value = "" ) with gr.Group(): with gr.Row(): prompt_field = gr.Textbox( label = "Prompt (77 token limit):", # label = "Prompt:", # info = "77 token limit", value = default_prompt ) with gr.Row( elem_id = "negative_prompt_field_row_id", visible = default_negative_prompt_field_row_visibility ): negative_prompt_field = gr.Textbox( label = "Negative Prompt (77 token limit):", # label = "Negative Prompt:", # info = "77 token limit", value = default_negative_prompt ) with gr.Row( elem_id = "negative_prompt_for_sdxl_turbo_field_row_id", visible = default_negative_prompt_for_sdxl_turbo_field_row_visibility ): negative_prompt_for_sdxl_turbo_field = gr.HTML( value = "
Negative prompt is not used for SDXL Turbo.
" ) with gr.Group( visible = refiner_group_visible ): refiner_label_text = "Refiner" with gr.Accordion( label = refiner_label_text, open = refiner_accordion_open, visible = refiner_accordion_visible ) as refiner_accordion: with gr.Row(): refining_selection_field = gr.Radio( label = "Use refiner?", choices = ["Yes", "No"], value = default_refine_option, container = True ) with gr.Row(): refining_denoise_start_field = gr.Slider( label = "Refiner denoise start %", minimum = minimum_refiner_denoise_start, maximum = maximum_refiner_denoise_start, value = default_refiner_denoise_start, step = refiner_denoise_start_input_slider_steps ) with gr.Row( elem_id = "refining_use_denoising_start_in_base_model_when_using_refiner_field_row_id", visible = default_refining_use_denoising_start_in_base_model_when_using_refiner_field_row_visibility ): refining_use_denoising_start_in_base_model_when_using_refiner_field = gr.Checkbox( label = "Use \"denoising_start\" value as \"denoising_end\" value in base model generation when using refiner", value = default_use_denoising_start_in_base_model_when_using_refiner_is_selected, interactive = True, container = True, elem_classes = "sp_checkbox" ) with gr.Row(): refining_base_model_output_to_refiner_is_in_latent_space_field = gr.Checkbox( label = "Base model output in latent space instead of PIL image when using refiner", value = default_base_model_output_to_refiner_is_in_latent_space_is_selected, interactive = True, container = True, elem_classes = "sp_checkbox" ) # Might add this again eventually (not needed, but to match older configuration) model_configuration_include_refiner_number_of_steps_object = {} refining_steps_for_older_configuration_field_visible = False if default_model_configuration in model_configuration_include_refiner_number_of_steps_object: refining_steps_for_older_configuration_field_visible = True with gr.Row( elem_id = "refining_steps_for_older_configuration_field_row_id", visible = refining_steps_for_older_configuration_field_visible ): refining_steps_for_older_configuration_field = gr.Slider( label = "Refining steps:", minimum = 1, maximum = maximum_refining_steps_for_online_config_field, value = default_refining_steps_for_online_config_field, step = 1 ) with gr.Group( visible = upscaler_group_visible ): with gr.Accordion( label = "Upscaler", open = upscaler_accordion_open, visible = upscaler_group_visible ): # # # # Upscaler # # # with gr.Row(): upscaling_selection_field = gr.Radio( label = "Upscale by 2x?", choices = ["Yes", "No"], value = default_upscale_option, container = True ) with gr.Row(): upscaling_num_inference_steps_field = gr.Slider( label = "Upscaler number of steps", minimum = 1, maximum = maximum_upscaler_steps, value = default_upscaler_steps, step = 1 ) if ( (enable_refiner == 1) or (enable_upscaler == 1) ): refiner_and_upscaler_text_field = gr.HTML( value = "
" + default_refiner_and_upscaler_status_text + "
" ) with gr.Column( scale = 2, min_width = 200 ): with gr.Group(): with gr.Row(): base_model_field = gr.Dropdown( label = "Model:", choices = default_base_model_choices_array, value = default_base_model_nicely_named_value, type = "index", filterable = False, interactive = True, elem_classes = "sp_dropdown" ) model_configuration_dropdown_field_values_for_js = "" model_configuration_dropdown_fields_array = [] for this_base_model in base_model_array: this_model_configuration_choices_array = [] for this_model_configuration in base_model_object_of_model_configuration_arrays[this_base_model]: this_model_configuration_choices_array.append( model_configuration_names_object[this_model_configuration] ) this_configuration_field_row_visibility = False if ( (this_base_model == default_base_model) and (allow_other_model_versions == 1) ): this_configuration_field_row_visibility = True this_configuration_field_default_value = model_configuration_names_object[base_model_model_configuration_defaults_object[this_base_model]] this_configuration_field_default_value_for_js = this_configuration_field_default_value this_configuration_field_default_value_for_js = this_configuration_field_default_value_for_js.replace("\"", "\\\"") model_configuration_dropdown_field_values_for_js += "\"" + this_base_model + "\": \"" + this_configuration_field_default_value_for_js + "\"," with gr.Row( elem_id = "model_configuration_field_" + this_base_model + "_row_id", visible = this_configuration_field_row_visibility ): this_configuration_field = gr.Dropdown( label = "Version:", choices = this_model_configuration_choices_array, value = this_configuration_field_default_value, type = "index", filterable = False, interactive = True, elem_classes = "sp_dropdown" ) model_configuration_dropdown_fields_array.append(this_configuration_field) with gr.Row(): scheduler_field = gr.Dropdown( label = "Scheduler / Sampler:", choices = default_scheduler_choices_array, value = default_scheduler_nicely_named_value, type = "index", filterable = False, interactive = True, elem_classes = "sp_dropdown" ) with gr.Row(): image_width_field = gr.Slider( label = "Width:", minimum = minimum_width, maximum = maximum_width, value = default_width, step = width_and_height_input_slider_steps, interactive = True ) image_height_field = gr.Slider( label = "Height:", minimum = minimum_height, maximum = maximum_height, value = default_height, step = width_and_height_input_slider_steps, interactive = True ) with gr.Row( elem_id = "base_model_steps_field_row_id", visible = default_base_model_steps_field_row_visibility ): base_model_steps_field = gr.Slider( label = "Steps:", minimum = 1, maximum = maximum_base_model_steps, value = default_base_model_steps, step = 1, interactive = True ) with gr.Row( elem_id = "base_model_steps_field_for_sdxl_turbo_field_row_id", visible = default_base_model_steps_field_for_sdxl_turbo_field_row_visibility ): base_model_steps_field_for_sdxl_turbo_field = gr.Slider( label = "Steps:", info = "Try using only 1 or a couple of steps.", minimum = 1, maximum = maximum_base_model_steps_for_sdxl_turbo, value = default_base_model_steps_for_sdxl_turbo, step = 1, interactive = True ) with gr.Row( elem_id = "guidance_scale_field_row_id", visible = default_guidance_scale_field_row_visibility ): guidance_scale_field = gr.Slider( label = "Guidance Scale:", minimum = minimum_guidance_scale, maximum = maximum_guidance_scale, value = default_guidance_scale, step = guidance_scale_input_slider_steps, interactive = True ) with gr.Row( elem_id = "guidance_scale_for_sdxl_turbo_field_row_id", visible = default_guidance_scale_for_sdxl_turbo_field_row_visibility ): guidance_scale_for_sdxl_turbo_field = gr.HTML( value = "
Guidance scale is not used for SDXL Turbo.
" ) with gr.Row(): if default_seed_value == "random": default_seed_value = generate_random_seed() # If you use a slider or number field for the seed, some # seeds can't be duplicated using those fields. If you # enter a number greater than 9007199254740992, the seed # won't reliably be used. This is a technical limitation # as of writing this. See the bug report here: # https://github.com/gradio-app/gradio/issues/5354 # # Until this is fixed, we use a textbox if the max seed # allowed is greater than that number. Using the slider, # and not entering a number, might be the way to get # reliable numbers above that number, if you just don't # then use the up and down arrows in the field to go up # or down a number. # # For now, I do this, but I might eventually have a # setting on the page to allow the slider. if make_seed_selection_a_textbox == 1: seed_field = gr.Textbox( label = "Seed:", value = default_seed_value, interactive = True ) else: seed_field = gr.Slider( label = "Seed:", minimum = 0, maximum = maximum_seed, value = default_seed_value, step = 1, interactive = True ) with gr.Column( scale = 3, min_width = 200 ): output_image_field_visibility = True output_image_gallery_field_visibility = False if use_image_gallery == 1: output_image_field_visibility = False output_image_gallery_field_visibility = True with gr.Row( visible = output_image_field_visibility ): output_image_field = gr.Image( label = "Generated Image", type = "pil", height = gradio_image_component_height, elem_classes = "image_scaling" ) with gr.Row( visible = output_image_gallery_field_visibility ): output_image_gallery_field = gr.Gallery( label = "Generated Images", value = [], selected_index = 0, allow_preview = True, preview = True, # columns = "2", # rows = None, # columns = "2", # rows = None, height = gradio_image_gallery_component_height, object_fit = "scale-down", show_download_button = True, elem_classes = "image_scaling" ) with gr.Row( visible = False ) as output_image_preview_field_row: output_image_preview_field_every_value = None if enable_image_preview == 1: output_image_preview_field_every_value = load_image_preview_frequency_in_seconds output_image_preview_field = gr.Image( elem_id = "image_preview_id", # value = load_image_preview(user_id_state), # every = output_image_preview_field_every_value, label = "Preview", type = "pil", interactive = False, show_download_button = True, height = gradio_image_gallery_component_height, elem_classes = "image_scaling" ) with gr.Row(): error_text_field = gr.Textbox( label = "Error Information:", value = "", show_copy_button = True, lines = 5, max_lines = 8, autoscroll = False, interactive = False, container = True, elem_classes = "textbox_vertical_scroll", visible = False ) with gr.Row(): output_text_field = gr.Textbox( label = "Prompt Information:", value = "After an image is generated, its generation information will appear here. All of this information is also embedded in the image itself. If you open the image in a text program, it will appear at the top." + additional_prompt_info_html, show_copy_button = True, lines = 5, max_lines = 8, autoscroll = False, interactive = False, container = True, elem_classes = "textbox_vertical_scroll" ) with gr.Group( visible = False ) as prompt_truncated_field_group: with gr.Row(): prompt_truncated_field = gr.Textbox( label = "Prompt Truncated:", info = "", show_copy_button = True, lines = 3, max_lines = 5, autoscroll = False, interactive = False, container = True, elem_classes = "textbox_vertical_scroll" ) with gr.Group( visible = False ) as negative_prompt_truncated_field_group: with gr.Row(): negative_prompt_truncated_field = gr.Textbox( label = "Negative Prompt Truncated:", info = "Your negative prompt was been truncated because it was too long. The part below was removed.", value = "", show_copy_button = True, lines = 3, max_lines = 5, autoscroll = False, interactive = False, container = True, elem_classes = "textbox_vertical_scroll" ) if enable_close_command_prompt_button == 1: with gr.Row(): close_command_prompt_button = gr.Button( value = "Close Command Prompt", variant = "stop" ) gr.Markdown("Closing the command prompt will cancel any images in the process of being created. You will need to launch it again, and then likely refresh the page, to create more images.") with gr.Accordion( label = "Other Settings", open = True ): with gr.Row(): add_seed_into_pipe_field = gr.Checkbox( label = "Add seed to generation (to make it deterministic)", value = default_add_seed_into_pipe_is_selected, interactive = True, container = False, elem_classes = "sp_checkbox" ) with gr.Row(): use_torch_manual_seed_but_do_not_add_to_pipe_field = gr.Checkbox( label = "Use torch.manual_seed, but don't actually add seed to generation (ignored if \"Add seed to generation\" is checked. This is to maintain compatibility with PhotoReal site)", value = default_use_torch_manual_seed_but_do_not_add_to_pipe_is_selected, interactive = True, container = False, elem_classes = "sp_checkbox" ) with gr.Row(): save_or_display_word_text_for_save_base_image = "Display" if auto_save_imagery == 1: save_or_display_word_text_for_save_base_image = "Save" save_base_image_when_using_refiner_field = gr.Checkbox( label = save_or_display_word_text_for_save_base_image + " base image as well when using refiner (doesn't work yet)", value = default_save_base_image_when_using_refiner_is_selected, interactive = True, container = False, elem_classes = "sp_checkbox" ) with gr.Row(): save_or_display_word_text_for_save_refined_image = "display" if auto_save_imagery == 1: save_or_display_word_text_for_save_refined_image = "save" save_refined_image_when_using_upscaler_field = gr.Checkbox( label = "If applicable, " + save_or_display_word_text_for_save_refined_image + " refined image as well when using upscaler (doesn't work yet)", value = default_save_refined_image_when_using_upscaler_is_selected, interactive = True, container = False, elem_classes = "sp_checkbox" ) if len(ending_html) > 0: with gr.Accordion( label = "Information", open = True ): gr.Markdown(ending_html) ##################### # # Update Refiner and Upscaler Status Function for Javascript # # When the refiner or upscaler is turned on or off, a text message is # printed on the page. That needs to be updated. # ##################### update_refiner_and_upscaler_status_function_js = """ async ( refiningSelectionFieldValue, upscalingSelectionFieldValue ) => {{ "use strict"; var refinerOnText = "{0}"; var refinerOffText = "{1}"; var upscalerOnText = "{2}"; var upscalerOffText = "{3}"; var refinerAndUpscalerInfoMessageHtml = ""; if (refiningSelectionFieldValue === "Yes") {{ refinerAndUpscalerInfoMessageHtml += refinerOnText; }} else {{ refinerAndUpscalerInfoMessageHtml += refinerOffText; }} if (upscalingSelectionFieldValue === "Yes") {{ refinerAndUpscalerInfoMessageHtml += upscalerOnText; }} else {{ refinerAndUpscalerInfoMessageHtml += upscalerOffText; }} document.getElementById("refiner_and_upscaler_info_message_div_id").innerHTML = refinerAndUpscalerInfoMessageHtml; }} """.format( refiner_on_text, refiner_off_text, upscaler_on_text, upscaler_off_text ) ##################### # # Model Change Function for Javascript # # When the base model or model configuration is changed, we may need # to show and hide certain fields. # ##################### model_change_function_js = """ async ( baseModelFieldFullNameValue, possiblyModelConfigurationFullNameValue ) => {{ "use strict"; var baseModelNamesObject = {0}; var modelConfigurationNamesObject = {1}; var baseModelArray = {2}; var modelConfigurationIncludeRefinerNumberOfStepsObject = {3}; var baseModelsNotSupportingDenoisingEndForBaseModelObject = {4} var allowOtherModelVersions = {5}; var baseModelFullNamesToBaseModelIdConversion = {{}}; Object.keys(baseModelNamesObject).forEach(key => {{ baseModelFullNamesToBaseModelIdConversion[baseModelNamesObject[key]] = key; }}); var baseModelFieldValue = ""; if (baseModelFullNamesToBaseModelIdConversion.hasOwnProperty(baseModelFieldFullNameValue)) {{ baseModelFieldValue = baseModelFullNamesToBaseModelIdConversion[baseModelFieldFullNameValue]; }} var modelConfigurationFullNameValue = "" var isBaseModelDropdownChange = 0 if (baseModelFieldFullNameValue === possiblyModelConfigurationFullNameValue) {{ isBaseModelDropdownChange = 1; modelConfigurationFullNameValue = window.modelConfigurationDropdownFieldValuesObject[baseModelFieldValue]; }} else {{ modelConfigurationFullNameValue = possiblyModelConfigurationFullNameValue; window.modelConfigurationDropdownFieldValuesObject[baseModelFieldValue] = modelConfigurationFullNameValue; }} var modelConfigurationFullNamesToModelConfigurationIdConversion = {{}}; Object.keys(modelConfigurationNamesObject).forEach(key => {{ modelConfigurationFullNamesToModelConfigurationIdConversion[modelConfigurationNamesObject[key]] = key; }}); var modelConfigurationNameValue = ""; if (modelConfigurationFullNamesToModelConfigurationIdConversion.hasOwnProperty(modelConfigurationFullNameValue)) {{ modelConfigurationNameValue = modelConfigurationFullNamesToModelConfigurationIdConversion[modelConfigurationFullNameValue]; }} for (var thisBaseModel of baseModelArray) {{ var thisModelConfigurationElementId = "model_configuration_field_" + thisBaseModel + "_row_id"; var thisModelConfigurationElementDisplay = "none"; if ( (thisBaseModel === baseModelFieldValue) && (allowOtherModelVersions === 1) ) {{ thisModelConfigurationElementDisplay = "block"; }} document.getElementById(thisModelConfigurationElementId).style.display = thisModelConfigurationElementDisplay; }} var modelConfigurationFullNamesToModelConfigurationIdConversion = {{}}; Object.keys(modelConfigurationNamesObject).forEach(key => {{ modelConfigurationFullNamesToModelConfigurationIdConversion[modelConfigurationNamesObject[key]] = key; }}); var modelConfigurationNameValue = ""; if (modelConfigurationFullNamesToModelConfigurationIdConversion.hasOwnProperty(modelConfigurationFullNameValue)) {{ modelConfigurationNameValue = modelConfigurationFullNamesToModelConfigurationIdConversion[modelConfigurationFullNameValue]; }} if ( baseModelFieldValue && modelConfigurationNameValue ) {{ var negativePromptFieldDisplay = "block"; var negativePromptForSdxlTurboFieldDisplay = "none"; var baseModelNumInferenceStepsFieldDisplay = "block"; var baseModelNumInferenceStepsFieldForSdxlTurboFieldDisplay = "none"; var guidanceScaleFieldDisplay = "block"; var guidanceScaleForSdxlTurboFieldDisplay = "none"; if (baseModelFieldValue === "sdxl_turbo") {{ negativePromptFieldDisplay = "none"; negativePromptForSdxlTurboFieldDisplay = "block"; baseModelNumInferenceStepsFieldDisplay = "none"; baseModelNumInferenceStepsFieldForSdxlTurboFieldDisplay = "block"; guidanceScaleFieldDisplay = "none"; guidanceScaleForSdxlTurboFieldDisplay = "block"; }} document.getElementById("negative_prompt_field_row_id").style.display = negativePromptFieldDisplay; document.getElementById("negative_prompt_for_sdxl_turbo_field_row_id").style.display = negativePromptForSdxlTurboFieldDisplay; document.getElementById("base_model_steps_field_row_id").style.display = baseModelNumInferenceStepsFieldDisplay; document.getElementById("base_model_steps_field_for_sdxl_turbo_field_row_id").style.display = baseModelNumInferenceStepsFieldForSdxlTurboFieldDisplay; document.getElementById("guidance_scale_field_row_id").style.display = guidanceScaleFieldDisplay; document.getElementById("guidance_scale_for_sdxl_turbo_field_row_id").style.display = guidanceScaleForSdxlTurboFieldDisplay; var refiningStepsForOlderConfiguration = "none"; if (Object.keys(modelConfigurationIncludeRefinerNumberOfStepsObject).includes(modelConfigurationNameValue)) {{ refiningStepsForOlderConfiguration = "block"; }} var refiningUseDenoisingStartInBaseModelWhenUsingRefinerFieldDisplay = "block"; if (Object.keys(baseModelsNotSupportingDenoisingEndForBaseModelObject).includes(baseModelFieldValue)) {{ refiningUseDenoisingStartInBaseModelWhenUsingRefinerFieldDisplay = "none"; }} document.getElementById("refining_use_denoising_start_in_base_model_when_using_refiner_field_row_id").style.display = refiningUseDenoisingStartInBaseModelWhenUsingRefinerFieldDisplay; document.getElementById("refining_steps_for_older_configuration_field_row_id").style.display = refiningStepsForOlderConfiguration; }} }} """.format( base_model_names_object, model_configuration_names_object, base_model_array, model_configuration_include_refiner_number_of_steps_object, base_models_not_supporting_denoising_end_for_base_model_object, allow_other_model_versions ) base_model_field.change( fn = None, inputs = [ base_model_field ], outputs = None, js = model_change_function_js ) for this_model_configuration_dropdown_field in model_configuration_dropdown_fields_array: this_model_configuration_dropdown_field.change( fn = None, inputs = [ base_model_field, this_model_configuration_dropdown_field ], outputs = None, js = model_change_function_js ) output_image_gallery_field.select( fn = update_prompt_info_from_gallery, inputs = [ prompt_information_array_state ], outputs = [ output_image_gallery_field, output_text_field ], show_progress = "hidden" ) if ( (enable_refiner == 1) or (enable_upscaler == 1) ): triggers_array = [] if enable_refiner == 1: triggers_array.extend([ base_model_field.change, refining_selection_field.change ]) for this_model_configuration_dropdown_field in model_configuration_dropdown_fields_array: triggers_array.extend([ this_model_configuration_dropdown_field.change ]) if enable_upscaler == 1: triggers_array.extend([ upscaling_selection_field.change ]) gr.on( triggers = triggers_array, fn = None, inputs = [ refining_selection_field, upscaling_selection_field ], outputs = None, show_progress = "hidden", queue = False, js = update_refiner_and_upscaler_status_function_js ) create_image_function_inputs = [ base_model_field, prompt_field, negative_prompt_field, scheduler_field, image_width_field, image_height_field, guidance_scale_field, base_model_steps_field, base_model_steps_field_for_sdxl_turbo_field, seed_field, add_seed_into_pipe_field, use_torch_manual_seed_but_do_not_add_to_pipe_field, refining_selection_field, refining_denoise_start_field, refining_use_denoising_start_in_base_model_when_using_refiner_field, refining_base_model_output_to_refiner_is_in_latent_space_field, refining_steps_for_older_configuration_field, upscaling_selection_field, upscaling_num_inference_steps_field, image_gallery_array_state, prompt_information_array_state, last_model_configuration_name_selected_state, last_refiner_name_selected_state, last_upscaler_name_selected_state, stored_pipe_state, stored_refiner_state, stored_upscaler_state, save_base_image_when_using_refiner_field, save_refined_image_when_using_upscaler_field, user_id_state, image_generation_id_state, canceled_images_array_state ] for this_model_configuration_dropdown_field in model_configuration_dropdown_fields_array: create_image_function_inputs.append( this_model_configuration_dropdown_field ) generate_image_button_click_event = generate_image_button.click( fn = before_create_image_function, inputs = None, outputs = [ generate_image_button, output_image_field, output_image_gallery_field, output_text_field, output_image_preview_field_row, prompt_truncated_field_group, prompt_truncated_field, negative_prompt_truncated_field_group, negative_prompt_truncated_field, cancel_image_button_row, cancel_image_button, cancel_image_message_field_row, cancel_image_message_field, error_text_field, image_generation_id_state ], show_progress = "hidden", queue = True ).then( fn = create_image_function, inputs = create_image_function_inputs, outputs = [ output_image_field, output_image_gallery_field, output_text_field, prompt_truncated_field_group, prompt_truncated_field, negative_prompt_truncated_field_group, negative_prompt_truncated_field, last_model_configuration_name_selected_state, last_refiner_name_selected_state, last_upscaler_name_selected_state, stored_pipe_state, stored_refiner_state, stored_upscaler_state, error_text_field ], show_progress = "full", queue = True ).then( fn = after_create_image_function, inputs = None, outputs = [ generate_image_button, output_image_field, output_image_gallery_field, output_text_field, output_image_preview_field_row, generate_image_button_row, cancel_image_button_row, cancel_image_button, cancel_image_message_field_row, cancel_image_message_field ], show_progress = "hidden", queue = True ) verify_seed_field_textbox_function_js = """ async ( seedFieldTextboxValue ) => {{ "use strict"; var defaultSeedMaximum = parseInt({0}); seedFieldTextboxValue = parseInt(seedFieldTextboxValue); if (isNaN(seedFieldTextboxValue)) {{ seedFieldTextboxValue = ""; }} else if (seedFieldTextboxValue > defaultSeedMaximum) {{ seedFieldTextboxValue = defaultSeedMaximum; }} return [ seedFieldTextboxValue ]; }} """.format( maximum_seed ) if make_seed_selection_a_textbox == 1: seed_field.change( fn = None, inputs = [ seed_field ], outputs = [ seed_field ], show_progress = "hidden", queue = False, js = verify_seed_field_textbox_function_js ) if enable_image_generation_cancellation == 1: cancel_image_click_event = cancel_image_button.click( fn = cancel_image_function, inputs = [ user_id_state, image_generation_id_state ], outputs = [ generate_image_button_row, generate_image_button, output_text_field, output_image_preview_field_row, cancel_image_button_row, cancel_image_button, cancel_image_message_field_row, cancel_image_message_field ], show_progress = "hidden", cancels = [generate_image_button_click_event], queue = True ) if enable_close_command_prompt_button == 1: # https://github.com/gradio-app/gradio/pull/2433/files close_command_prompt_button_click_event = close_command_prompt_button.click( fn = close_command_prompt, inputs = None, outputs = None, cancels = [generate_image_button_click_event], queue = True ) # Remove last comma model_configuration_dropdown_field_values_for_js = model_configuration_dropdown_field_values_for_js[:-1] script_on_load_js = """ async () => {{ "use strict"; window.modelConfigurationDropdownFieldValuesObject = {{{0}}}; document.querySelectorAll(".textbox_vertical_scroll textarea").forEach(e => e.classList.remove("scroll-hide")); }} """.format( model_configuration_dropdown_field_values_for_js ) model_base_model_and_model_configuration_inputs = [ base_model_field, initial_model_configuration_name_selected_state ] model_base_model_and_model_configuration_outputs = [] for this_model_configuration_dropdown_field in model_configuration_dropdown_fields_array: model_base_model_and_model_configuration_inputs.append( this_model_configuration_dropdown_field ) model_base_model_and_model_configuration_outputs.append( this_model_configuration_dropdown_field ) sd_interface_load_outputs = create_image_function_inputs + [ initial_model_configuration_name_selected_state, generate_image_button ] every_value_in_seconds_for_image_preview = None if enable_image_preview == 1: every_value_in_seconds_for_image_preview = load_image_preview_frequency_in_seconds sd_interface_continuous = sd_interface.load( fn = get_query_params, inputs = None, outputs = sd_interface_load_outputs, show_progress = "hidden", queue = False, scroll_to_output = False, js = script_on_load_js ).then( fn = set_base_model_and_model_configuration_from_query_params, inputs = model_base_model_and_model_configuration_inputs, outputs = model_base_model_and_model_configuration_outputs, show_progress = "hidden", queue = False ).then( fn = set_base_model_and_model_configuration_from_query_params, inputs = model_base_model_and_model_configuration_inputs, outputs = model_base_model_and_model_configuration_outputs, show_progress = "hidden", queue = False, ).then( fn = load_image_preview, inputs = [ user_id_state ], outputs = [ output_image_preview_field ], show_progress = "hidden", every = every_value_in_seconds_for_image_preview, queue = True ) sd_interface.queue( max_size = max_queue_size ) inbrowser = False if auto_open_browser == 1: inbrowser = True sd_interface.queue().launch( #sd_interface.launch( inbrowser = inbrowser, share = None, show_api = False, quiet = True, show_error = True, state_session_capacity = 10000, max_threads = 40 )