a AþdFã@sbddlZddlZddlZddlZddlmZddlmZdd„Zdd„Z dd „Z d d œd d „Z dS)éN)ÚThread)Ú startup_timerc Cst d¡ tj¡t d¡ dd„¡ddl}t d¡ddl}t d¡t j dt dd t j dt d d ddl }t d ¡dd lm}m}m}m}t d ¡ddl}t d¡ddl}t d¡ddlm} |  ¡t d¡ddlm} m} m} t d¡dS)Nztorch.distributed.nnÚxformerscSs d| ¡vS)Nz"A matching Triton is not available)Ú getMessage)Úrecord©rú:/var/www/html/stable-diffusion-webui/modules/initialize.pyÚ ózimports..rz import torchÚignoreÚpytorch_lightning©ÚactionÚcategoryÚmoduleÚ torchvisionz import gradio)ÚpathsÚtimerÚ import_hookÚerrorsz setup pathsz import ldmz import sgm)Ú shared_initzinitialize shared)Ú processingÚgradio_extensonsÚuiz other imports)ÚloggingÚ getLoggerÚsetLevelÚERRORÚ addFilterÚtorchrrr ÚwarningsÚfilterwarningsÚDeprecationWarningÚ UserWarningÚgradioÚmodulesrrrrZldm.modules.encoders.modulesZsgm.modules.encoders.modulesrÚ initializerrr) rr r$rrrrZldmZsgmrrrrrrrÚimports s*        r'cCs*ddlm}|js&ddlm}| ¡dS)Nr©Úcmd_opts)r)Úmodules.shared_cmd_optionsr)Úskip_version_checkr%rÚcheck_versions)r)rrrrr,)s  r,cCsÔddlm}| ¡| ¡| ¡| ¡| ¡ddlm}| ¡ddlm }|  ¡t   d¡ddl m}ddlm}tjdtd d |  |j¡t   d ¡dd lm}|  |j¡t   d ¡tdddS)Nr©Úinitialize_util©Ú modelloader©Ú sd_modelszsetup SD modelr()Úcodeformer_modelr z(torchvision.transforms.functional_tensorr zsetup codeformer)Ú gfpgan_modelz setup gfpganF©Úreload_script_modules)r%r.Úfix_torch_versionÚfix_asyncio_event_loop_policyÚvalidate_tls_optionsÚconfigure_sigint_handlerÚconfigure_opts_onchanger0Zcleanup_modelsr2Z setup_modelrrr*r)r3r r!r#Úcodeformer_models_pathr4Úgfpgan_models_pathÚinitialize_rest)r.r0r2r)r3r4rrrr&1s(           r&Fr5csnddlm}ddlm}| ¡t d¡ddlm}| ¡t d¡ddlm }|  ¡t d¡dd lm ‰m }m }|jr˜| ¡jˆ_| ¡d Sdd lm}| ¡t d ¡dd lm}| |j¡t d¡t d¡| ¡Wd ƒn1s0Y|rBdd„tj ¡DƒD]} t | ¡q&t d¡ddlm} |  ¡t d¡ddlm} |   ¡t d¡ddlm!} | j! "¡t d¡ddlm#} m$}m%‰|  &|j'¡ˆ '¡t d¡ddlm(}| )¡t d¡‡‡fdd„}t*|d +¡dd lm,}| -¡t d!¡dd"lm.}| /¡| 0¡dd#lm1}| /¡| 2¡t d$¡d S)%zE Called both from initialize() and when reloading the webui. rr()Ú sd_samplersz set samplers)Ú extensionszlist extensionsr-zrestore config state file)ÚsharedÚupscalerÚscriptsNr1zlist SD models)Ú localizationzlist localizationsz load scriptscSsg|]\}}| d¡r|‘qS)z modules.ui)Ú startswith)Ú.0ÚnamerrrrÚ rr z#initialize_rest..zreload script modulesr/zload upscalers)Úsd_vaez refresh VAE)Útextual_inversionz#refresh textual inversion templates)Úscript_callbacksÚsd_hijack_optimizationsÚ sd_hijackzscripts list_optimizers)Úsd_unetzscripts list_unetscs0ˆjˆjdurˆ ¡ddlm}| ¡dS)a, Accesses shared.sd_model property to load model. After it's available, if it has been loaded before this access by some extension, its optimization may be None because the list of optimizaers has neet been filled by that time, so we apply optimization again. Nr©Údevices)Úsd_modelZcurrent_optimizerZapply_optimizationsr%rPZfirst_time_calculationrO©rMrArrÚ load_model‹s   z#initialize_rest..load_model)Útarget)Ú shared_itemszreload hypernetworks)Úui_extra_networks)Úextra_networkszinitialize extra networks)3r*r)r%r?Z set_samplersrrr@Úlist_extensionsr.Úrestore_config_state_filerArBrCÚ ui_debug_modeZUpscalerLanczosZscalersZ sd_upscalersZ load_scriptsr2Z list_modelsrDZlist_localizationsÚlocalizations_dirÚ subcategoryÚsysÚitemsÚ importlibÚreloadr0Zload_upscalersrIZrefresh_vae_listrJZ list_textual_inversion_templatesrKrLrMZon_list_optimizersZlist_optimizersrNZ list_unetsrÚstartrUZreload_hypernetworksrVr&Zregister_default_pagesrWZregister_default_extra_networks)r6r)r?r@r.rBrCr2rDrr0rIrJrKrLrNrSrUrVrWrrRrr>Nsn              (                r>) r_rr]r Ú threadingrÚ modules.timerrr'r,r&r>rrrrÚs