a Ad@sFddlZddlmZmZddlmZddlmZGdddej Z dS)N)scripts sd_models)create_refresh_button)InputAccordionc@s<eZdZdZdZddZddZddZd d Zd d Z d S) ScriptRefiner accordionsFcCsdSNselfr r J/var/www/html/stable-diffusion-webui/modules/processing_scripts/refiner.py__init__ szScriptRefiner.__init__cCsdSNRefinerr r r r r titleszScriptRefiner.titlecCstjSr)r AlwaysVisible)r is_img2imgr r r showszScriptRefiner.showc stdd|dd}tftjd|dtddd }t|tjd d |d tj d dddd|ddd}Wdn1s0YWdn1s0Ydd|dd f|fdd f|dfg|_ |||fS)NFrenable)labelelem_id Checkpoint checkpointz3switch to another model in the middle of generation)rrchoicesvaluetooltipcSs dtiS)Nr)rcheckpoint_tilesr r r r z"ScriptRefiner.ui..Zcheckpoint_refreshg?z Switch atg{Gz?g?Z switch_atzzfraction of sampling steps when the switch to refiner model should happen; 1=never, 0.5=switch in the middle of generation)rrminimummaximumsteprrcSst|}|durdS|jSr)rget_closet_checkpoint_matchr)rinfor r r lookup_checkpoints z+ScriptRefiner.ui..lookup_checkpointcSsd|vSrr dr r r r"rcs|dSr)getr&r%r r r#rzRefiner switch at) rrgrRowDropdownrrr list_modelsSliderinfotext_fields)r renable_refinerrefiner_checkpointrefiner_switch_atr r)r uis Z zScriptRefiner.uicCs*|r |dvrd|_d|_n ||_||_dS)N)NrNone)r1r2)r pr0r1r2r r r setup)s  zScriptRefiner.setupN) __name__ __module__ __qualname__section create_groupr rrr3r6r r r r rsr) gradior*modulesrrZmodules.ui_commonrZmodules.ui_componentsrScriptBuiltinUIrr r r r s