a Ad @@sddlZddlZddlmZddlmZmZmZddlm Z ddl m Z ddl m Z mZddZGd d d ZGd d d ZGd ddZGdddZGdddZGdddZGdddZedddgZeggggggggggggggggggggdZddZee e dddZd d!Zd"d#Zd$d%Zed&d'd(Zd)d*Z ed&d+d,Z!ed&d-d.Z"ed&d/d0Z#ed&d1d2Z$ed&d3d4Z%ed&d5d6Z&d7d8Z'd9d:Z(ed&d;d<Z)e*ee*efd=d>d?Z+d@dAZ,dBdCZ-dDdEZ.dFdGZ/dHdIZ0dJdKZ1dLdMZ2dNdOZ3dPdQZ4dRdSZ5dTdUZ6dVdWZ7dXdYZ8dZd[Z9d\d]Z:d^d_Z;d`daZdfdgZ?dhdiZ@djdkZAdldmZBdndoZCdpdqZDdrdsZEdtduZFdS)vN) namedtuple)OptionalDictAny)FastAPI)Blocks)errorstimercCs tjd|d|jdddS)NzError executing callback z for T)exc_info)rreportscript)cjobr@/var/www/html/stable-diffusion-webui/modules/script_callbacks.pyreport_exception src@seZdZddZdS)ImageSaveParamscCs||_||_||_||_dSN)imagepfilenamepnginfo)selfrrrrrrr__init__s zImageSaveParams.__init__N__name__ __module__ __qualname__rrrrrrsrc@seZdZddZdS)ExtraNoiseParamscCs||_||_||_dSr)noisexxi)rrr r!rrrr szExtraNoiseParams.__init__Nrrrrrrsrc@seZdZddZdS)CFGDenoiserParamscCs.||_||_||_||_||_||_||_dSr)r image_condsigma sampling_steptotal_sampling_steps text_cond text_uncond)rr r#r$r%r&r'r(rrrr,szCFGDenoiserParams.__init__Nrrrrrr"+sr"c@seZdZddZdS)CFGDenoisedParamscCs||_||_||_||_dSr)r r%r& inner_model)rr r%r&r*rrrrDs zCFGDenoisedParams.__init__Nrrrrrr)Csr)c@seZdZddZdS)AfterCFGCallbackParamscCs||_||_||_dSr)r r%r&)rr r%r&rrrrSszAfterCFGCallbackParams.__init__Nrrrrrr+Rsr+c@seZdZddZdS)UiTrainTabParamscCs ||_dSr)txt2img_preview_params)rr-rrrr_szUiTrainTabParams.__init__Nrrrrrr,^sr,c@seZdZddZdS)ImageGridLoopParamscCs||_||_||_dSr)imgscolsrows)rr/r0r1rrrrdszImageGridLoopParams.__init__Nrrrrrr.csr.ScriptCallbackr callback)callbacks_app_startedcallbacks_model_loadedcallbacks_ui_tabscallbacks_ui_train_tabscallbacks_ui_settingscallbacks_before_image_savedcallbacks_image_savedcallbacks_extra_noisecallbacks_cfg_denoisercallbacks_cfg_denoisedcallbacks_cfg_after_cfgcallbacks_before_componentcallbacks_after_componentcallbacks_image_gridcallbacks_infotext_pastedcallbacks_script_unloadedcallbacks_before_uicallbacks_on_reloadcallbacks_list_optimizerscallbacks_list_unetscCstD] }|qdSr) callback_mapvaluesclear) callback_listrrrclear_callbackss rL)demoappc CsVtdD]H}z&|||tjtj|jWqt yNt |dYq0qdS)Nr4app_started_callback) rHr3r startup_timerrecordospathbasenamer Exceptionr)rMrNr rrrrOs    rOc Cs<tdD].}z |Wqty4t|dYq0qdS)NrErHr3rUrr rrrapp_reload_callbacks    rXc Cs>tdD]0}z||Wqty6t|dYq0qdS)Nr5model_loaded_callbackrV)sd_modelr rrrrYs   rYc CsHg}tdD]6}z||pg7}Wq ty@t|dYq 0q |S)Nr6ui_tabs_callbackrVresr rrrr[s  r[)paramsc Cs>tdD]0}z||Wqty6t|dYq0qdS)Nr7rVr^r rrrui_train_tabs_callbacks   r`c Cs<tdD].}z |Wqty4t|dYq0qdS)Nr8ui_settings_callbackrVrWrrrras    rac Cs>tdD]0}z||Wqty6t|dYq0qdS)Nr9before_image_saved_callbackrVr_rrrrbs   rbc Cs>tdD]0}z||Wqty6t|dYq0qdS)Nr:image_saved_callbackrVr_rrrrcs   rcc Cs>tdD]0}z||Wqty6t|dYq0qdS)Nr;rVr_rrrextra_noise_callbacks   rdc Cs>tdD]0}z||Wqty6t|dYq0qdS)Nr<cfg_denoiser_callbackrVr_rrrres   rec Cs>tdD]0}z||Wqty6t|dYq0qdS)Nr=cfg_denoised_callbackrVr_rrrrfs   rfc Cs>tdD]0}z||Wqty6t|dYq0qdS)Nr>cfg_after_cfg_callbackrVr_rrrrgs   rgc KsFtdD]8}z|j|fi|Wqty>t|dYq0qdS)Nr?before_component_callbackrV componentkwargsr rrrrhs   rhc KsFtdD]8}z|j|fi|Wqty>t|dYq0qdS)Nr@after_component_callbackrVrirrrrls   rlc Cs>tdD]0}z||Wqty6t|dYq0qdS)NrAZ image_gridrVr_rrrimage_grid_callbacks   rm)infotextr^c Cs@tdD]2}z|||Wqty8t|dYq0qdS)NrBZinfotext_pastedrV)rnr^r rrrinfotext_pasted_callbacks   roc Cs@ttdD].}z |Wq ty8t|dYq 0q dS)NrCZscript_unloadedreversedrHr3rUrrWrrrscript_unloaded_callback s   rrc Cs@ttdD].}z |Wq ty8t|dYq 0q dS)NrDZ before_uirprWrrrbefore_ui_callbacks   rsc CsBg}tdD]0}z||Wq ty:t|dYq 0q |S)NrFlist_optimizersrVr\rrrlist_optimizers_callbacks  ruc CsBg}tdD]0}z||Wq ty:t|dYq 0q |S)NrG list_unetsrVr\rrrlist_unets_callback)s  rwcCs8ddtD}|r |djnd}|t||dS)NcSsg|]}|jtkr|qSrr__file__.0r rrr 6z add_callback..r unknown file)inspectstackrappendr2) callbacksfunrrrrr add_callback5srcsdddtD}|r |djnddkr0dStD]&}fdd|DD]}||qNq8dS)NcSsg|]}|jtkr|qSrrxrzrrrr|=r}z3remove_current_script_callbacks..rr~csg|]}|jkr|qSr)r r{cbrrrr|Br})rrrrHrIremove)rrKcallback_to_removerrrremove_current_script_callbacks<s rcs4tD]&}fdd|DD]}||qqdS)Ncsg|]}|jkr|qSrr3r callback_funcrrr|Hr}z1remove_callbacks_for_function..)rHrIr)rrKrrrrremove_callbacks_for_functionFs rcCsttd|dS)zregister a function to be called when the webui started, the gradio `Block` component and fastapi `FastAPI` object are passed as the argumentsr4NrrHrrrron_app_startedLsrcCsttd|dS)z@register a function to be called just before the server reloads.rENrrrrron_before_reloadRsrcCsttd|dS)zregister a function to be called when the stable diffusion model is created; the model is passed as an argument; this function is also called when the script is reloaded. r5Nrrrrron_model_loadedWsrcCsttd|dS)aregister a function to be called when the UI is creating new tabs. The function must either return a None, which means no new tabs to be added, or a list, where each element is a tuple: (gradio_component, title, elem_id) gradio_component is a gradio component to be used for contents of the tab (usually gr.Blocks) title is tab text displayed to user in the UI elem_id is HTML id for the tab r6Nrrrrr on_ui_tabs]s rcCsttd|dS)zregister a function to be called when the UI is creating new tabs for the train tab. Create your new tabs with gr.Tab. r7Nrrrrron_ui_train_tabsjsrcCsttd|dS)zregister a function to be called before UI settings are populated; add your settings by using shared.opts.add_option(shared.OptionInfo(...)) r8Nrrrrron_ui_settingsqsrcCsttd|dS)zregister a function to be called before an image is saved to a file. The callback is called with one argument: - params: ImageSaveParams - parameters the image is to be saved with. You can change fields in this object. r9Nrrrrron_before_image_savedwsrcCsttd|dS)zregister a function to be called after an image is saved to a file. The callback is called with one argument: - params: ImageSaveParams - parameters the image was saved with. Changing fields in this object does nothing. r:Nrrrrron_image_savedsrcCsttd|dS)zregister a function to be called before adding extra noise in img2img or hires fix; The callback is called with one argument: - params: ExtraNoiseParams - contains noise determined by seed and latent representation of image r;Nrrrrron_extra_noisesrcCsttd|dS)aregister a function to be called in the kdiffussion cfg_denoiser method after building the inner model inputs. The callback is called with one argument: - params: CFGDenoiserParams - parameters to be passed to the inner model and sampling state details. r<Nrrrrron_cfg_denoisersrcCsttd|dS)aregister a function to be called in the kdiffussion cfg_denoiser method after building the inner model inputs. The callback is called with one argument: - params: CFGDenoisedParams - parameters to be passed to the inner model and sampling state details. r=Nrrrrron_cfg_denoisedsrcCsttd|dS)aregister a function to be called in the kdiffussion cfg_denoiser method after cfg calculations are completed. The callback is called with one argument: - params: AfterCFGCallbackParams - parameters to be passed to the script for post-processing after cfg calculation. r>Nrrrrron_cfg_after_cfgsrcCsttd|dS)aregister a function to be called before a component is created. The callback is called with arguments: - component - gradio component that is about to be created. - **kwargs - args to gradio.components.IOComponent.__init__ function Use elem_id/label fields of kwargs to figure out which component it is. This can be useful to inject your own components somewhere in the middle of vanilla UI. r?Nrrrrron_before_components rcCsttd|dS)z`register a function to be called after a component is created. See on_before_component for more.r@Nrrrrron_after_componentsrcCsttd|dS)zregister a function to be called before making an image grid. The callback is called with one argument: - params: ImageGridLoopParams - parameters to be used for grid creation. Can be modified. rANrrrrr on_image_gridsrcCsttd|dS)zregister a function to be called before applying an infotext. The callback is called with two arguments: - infotext: str - raw infotext. - result: Dict[str, any] - parsed infotext parameters. rBNrrrrron_infotext_pastedsrcCsttd|dS)zregister a function to be called before the script is unloaded. Any hooks/hijacks/monkeying about that the script did should be reverted hererCNrrrrron_script_unloadedsrcCsttd|dS)z:register a function to be called before the UI is created.rDNrrrrr on_before_uisrcCsttd|dS)zregister a function to be called when UI is making a list of cross attention optimization options. The function will be called with one argument, a list, and shall add objects of type modules.sd_hijack_optimizations.SdOptimization to it.rFNrrrrron_list_optimizerssrcCsttd|dS)zregister a function to be called when UI is making a list of alternative options for unet. The function will be called with one argument, a list, and shall add objects of type modules.sd_unet.SdUnetOption to it.rGNrrrrr on_list_unetssr)GrrR collectionsrtypingrrrfastapirgradiormodulesrr rrrr"r)r+r,r.r2dictrHrLrOrXrYr[r`rarbrcrdrerfrgrhrlrmstrrorrrsrurwrrrrrrrrrrrrrrrrrrrrrrrrrrrs