nFjaUdZddlmZddlmZddlZddlZddlmZddl m Z ddl m Z dd l mZmZmZmZmZdd lmZd Zd Zd dgZgZded<eZded<eGddZddZddZddZddZ dd!Z!dd#Z"dd%Z#dd)Z$dd+Z%dd-Z&ddd1Z'ddd2Z(ddd6Z)dd7Z*ddd9Z+dd:Z,dd;Z-dd=Z.dd>Z/dd@Z0ddBZ1ddDZ2ddEZ3ddFZ4ddGZ5ddIZ6ddKZ7ddLZ8ddMZ9ddOZ:ddPZ;dddQZddVZ?d3dWddYZ@dZd3dd3dd[dd^ZAd3d_ddaZBdddbdcZCddZDdddeZEddfZFddhZGddiZHddkZIddlZJddmZKddnZLd3doddrZMddtZNdduZOddxZPddyZQdd|ZRGd}d~ZSdS)aTemporal upsampler plugin API. Temporal upsamplers are interchangeable handlers for frame interpolation. Handlers expose methods and supported multipliers through ``query_temporal_upsampler_def()`` and may expose config controls under ``wgp_config["temporal_upsamplers"][config_key]``. Definitions may also expose an optional ``description`` plus optional ``method_descriptions`` and ``method_parameters`` mappings for reusable discovery interfaces. Existing handlers without these fields remain valid. Discovery tests the existing optional ``enabled()`` method first. When it is absent, handlers may expose a ``status`` property containing ``"enabled"`` or ``"disabled"``. Discovery reports ``"unknown"`` only when neither contract provides a valid status. Disabled handlers may expose ``reason_disabled``. Plugin authors can register processors from ``plugin_info.json`` with: ```json { "temporal_upsampler_handlers": ["my_plugin.temporal.MyTemporalUpsampler"] } ``` ) annotations) dataclassN)Any)attention_shared_state)offload_registry)PROCESSOR_STATUS_DISABLEDPROCESSOR_STATUS_ENABLEDPROCESSOR_STATUS_UNKNOWNhandler_reason_disabledhandler_status)compatible_loaded_modeltemporal_upsamplers*zpostprocessing.dlss5.temporal_upsampler.DLSSGTemporalUpsampler list[Any]_temporal_upsampler_handlerszset[str],_registered_temporal_upsampler_handler_pathsc.eZdZUded<ded<ded<dS)TemporalUpsamplerConfigBindingrhandlerstr config_keyzlist[tuple[str, Any]]controlsN)__name__ __module__ __qualname____annotations__;C:\Users\johnn\Wan2GP\postprocessing\temporal_upsamplers.pyrr0s0LLLOOO######rrscalefloatreturnrct|}|rtt|n|dS)Ng)r" is_integerrintr!s r format_multiplierr)7s8 %LLE#..00 B3s5zz???llBrc&dt|S)Nx)r)r(s r format_multiplier_labelr,<s ) '' ) ))rmethodct|pdtt |SN)rstriplowerMULTIPLIER_SEPARATORr))r-r!s r format_multiplier_valuer4@sG&,B%%''--// a1E aGXY^G_G_ a aar default_scale float | Nonect|pd}t|pd}||sdS|t |d}|t r!|t t d}|sdSn t |vrdS t |p|S#t$rYdSwxYwr/)rr1r2 startswithlenr3r" ValueError)valuer-r5textsuffixs r parse_multiplier_suffixr>Ds  u{   ! ! # # ) ) + +D 2   $ $ & & , , . .F ??6 " "t #f++,, F -..011223 4   ' 'tV,}--- tts C11 C?>C?labelcnt|pdddt|S)Nr0z Upsampler )r removesuffixr,)r?r!s r format_method_scale_labelrCVs:%+2++L99 \ \z#_method_choices..es P P P vUFO P P Prmethods)getrTs r _method_choicesr_ds$ P PB1O1O P P PPrdict[str, str]c4dt|DS)Nci|]\}}|| SrrrYs r z"_method_labels..is L L LmeVFE L L Lr)r_r^s r _method_labelsrdhs L L{/K/K L L LLr fallback_namect|ddpd}|r|St|}|r|ddS|S)Nrr0rr)rr]r1r_r2)rTrermethod_choicess r _config_key_from_handler_defrhlss[__\266<"==CCEEJ$[11N$a ##     rhandler_moduleslist[str] | Nonedict[str, dict[str, Any]]cHi}|tn|D]}tt|pd}t |dsCt |}|sg|}||t||j <|S)Nr0default_config) temporal_upsampler_handlersrSrr1hasattrdictrmquery_temporal_upsampler_defrhr)risectionsrJ handler_clsconfigrTs r default_config_sectionsruvsH/>/F++O[[+C OO,A,A,C,CDD {$455  k002233  !>>@@ TZ-k;;OPPQQ Orc|tn|}|D]q}t|pd}|r |tvr1t t |||t|rdSr/)rnrr1rrIrSadd) server_config files_locatorrimodulesrJs r register_temporal_upsamplersr{s-<-D))/G??4:2$$&& tKKK #$?$9$$?$? }$]$]^^^488>>>> ??rF enabled_onlyboolc*fdtDS)Nc6g|]}rt||Sr)handler_enabled)rZrr|s r r[z2registered_temporal_upsamplers..s- p p p| pWfgnWoWo pG p p pr)rr|s`r registered_temporal_upsamplersrs p p p p#? p p pprcLt|d p|S)Nenabled)rorrHs r rrs$w ** * ?goo.?.??rlist[dict[str, Any]]c4dt|DS)Nc6g|]}|Sr)rq)rZrs r r[z1query_temporal_upsampler_defs..s$ o o owG 0 0 2 2 o o or)rrs r query_temporal_upsampler_defsrs o oB`amBnBn o o oorcz t|ddS#ttf$rYdSwxYw)Nposi)r"r] TypeErrorr:r^s r _handler_posrsJ[__UD11222 z "tts "%::c|di}t|tr1||vr- t||S#tt f$rYnwxYwt |S)N method_pos)r] isinstancerpr"rr:r)rTr-rs r _method_posrs|r22J*d##*(<(< F+,, ,:&    D   $ $$sAAA Any | NonectpdsdStfdtDdS)Nr0c3FK|]}||VdSrF) is_upsampling)rZrtemporal_upsamplings r z*find_temporal_upsampler..s7ssWH]H]^qHrHrsssssssr)rr1nextr)rs`r find_temporal_upsamplerrsQ "(b ) ) / / 1 1t ssss(Dsssuy z zzrct|pd}|sdStD]3}|dt|Dvr|cS4dS)Nr0cg|]\}}|SrrrZ_keys r r[z5find_temporal_upsampler_by_method..s```fac```r)rr1rr_rqr-rs r !find_temporal_upsampler_by_methodrs} 2   $ $ & &F t/ ``8\8\8^8^(_(_``` ` `NNN a 4rrcNt|}|td|d|S)Nz-No temporal upsampler registered for method '')r RuntimeErrorrs r $require_temporal_upsampler_by_methodrs3/77GT6TTTUUU Nrtuple[float, ...]ct|}|dSt|dit |dS)Nr multipliers)rtuplerqr]rrs r method_multipliersrs^/77Gr 5577;;M2NNRRSVW]S^S^`bcc d ddrlist[tuple[str, float]]c4dt|DS)Nc0g|]}t||fSr)r,)rZr!s r r[z1multiplier_choices_for_method..s& \ \ \ $U + +U 3 \ \ \r)r)r-s r multiplier_choices_for_methodrs \ \ASTZA[A[ \ \ \\rc@t|di|d}|sdS|dd}t|}|5||}||d|kr|d|vr|dS|dS)Nrrdefault_temporal_upsamplingr0rr)rr]r split_value)rTr-r default_valuersplits r _default_multiplier_from_defrs  r::>>vrJJKKK tOO$A2FFM%m44G##M22  qV!3!3aK8O8O8O q>rct|}|dSt|t|pdpdS)N@r0)rrrqrr1rs r default_multiplier_for_methodrsW/77Gs '(L(L(N(NPSTZT`^`PaPaPgPgPiPi j j qnqqrct|}|sdS t|}n&#ttf$rt |}YnwxYw||vr|nt |S)Nr)rr"rr:r)r-r!rs r normalize_multiplier_for_methodrs|$V,,K s6e  z "666-f556[((55.KF.S.SSs% AAtuple[str, float] | NonecRt|}|dn||SrF)rr)r;rs r split_temporal_upsampling_valuers+%e,,G?44(;(;E(B(BBr str | NonecTt|}|dn|||SrF)r build_value)r-r!rs r build_temporal_upsampling_valuers-/77G?44(;(;FE(J(JJrct|pd}|sdSt|}||St|d}|dn |j|}|p|S)Nr0r)rr1rrr)r;r<rr normalizeds r #normalize_temporal_upsampling_valuersw u{   ! ! # #D r +D 1 1E } /a99G .Ag.A5.IJ  rcBt|pd}|sdSt|}||S||}||S|\}}t ||}|rt||n|Sr/)rr1rrrdrqr]rC)r;r<rrr-r!r?s r format_temporal_upsampling_labelrs u{   ! ! # #D r%d++G    % %E } MFE 7??AA B B F Fv N NE6; E $UE 2 2 2Er1tuple[list[tuple[str, float]], float | None, str]ct|pd}|sgddfSt|}t||}||t ||pdfSr/)rr1rrr)r-r!multiplier_choicess r #normalize_temporal_upsampling_statersn 2   $ $ & &F 4|6v>> +FE : :E u&Efe&T&T&ZXZ ZZrcTt|}t||dn|dS)Nrr)rr)r- current_valuers r .normalize_temporal_upsampling_value_for_methodrs. +M : :E .vemssQVWXQY Z ZZrcg}t|D]4}||fdtDz }5dt|DS)Ncg|]J\}}t|t|pdt|pd||fKSr0)rrcasefold)rZr?r-rTs r r[z(query_method_choices..#syiii{HzBH[f55s5;B7G7G7P7P7R7RTWX^XdbdTeTeglntuiiirc"g|] \}}}}}||f Srr)rZrr?r-s r r[z(query_method_choices..$s' J J J 61aUFO J J Jr)rrqr_sorted)r|choicesrrTs @r query_method_choicesrsG1,??ii::<< iiiiL[\gLhLhiii i J J&// J J JJr field_helpc. t|dd}|g}i}|D]A\}}t|}|+||g||fB|D]\}}|} t | d|dd} t | ddpd d| g} r|  | dittrfd |D} | fd | Dz } |d | d |S) NTEMPORAL_UPSAMPLER_HELP_INTROzJTemporal upsamplers increase frame rate by generating intermediate frames.namer descriptionr0z### method_descriptionsc g|]K\}}t|t|dpdfLSr)rr]r1)rZr?r-rs r r[z*temporal_help_markdown..8sUyyy`m`egmE C(;(?(?(K(K(Qr$R$R$X$X$Z$Z[yyyrc6g|]\}}||k d|d|S)z- **z:** r)rZr?detailrs r r[z*temporal_help_markdown..9s@rrr]UFQWr\bfq\q\q0U0000\q\q\qrz ) rNr setdefaultrGitemsrqrr]r1rrpjoin)rgrintrorrhandler_choicesr?r-rrrTrlinesdetailsrrs @@r temporal_help_markdownr's J ?BN O OEwHO'LL v3F;;    & &w 3 3 : :E6? K K K+1133 , ,::<< ;??671:a=99::+//-<<BCCIIKK   & LL % % %)oo.CRHH )4 0 0 syyyyqxyyyG rrrrgrrr rE E**++++ ;;x  rtuple[str, str]c(dt||fS)NzTemporal Upsamplerr)r)rgrs r temporal_help_popupr>s !7S]!^!^!^ ^^rcdd|D}dtj|dddS)N|c3:K|]\}}t|VdSrF)r)rZ_labelr-s r rz#temporal_help_id..Cs,JJVJJJJJJrtemporal_upsampling_zutf-8 )rhashlibsha1encode hexdigest)rg signatures r temporal_help_idrBs`JJ>JJJJJI \',y/?/?/H/H"I"I"S"S"U"UVYWYVY"Z \ \\rlate_postprocessingrct|pd\}}dgt| z}|d|Dvrd}t||\}}}|||||dS)N)r0r)rDr0rch|]\}}|Srr)rZrr;s r z!dropdown_state..Js;;;5e;;;rr0)r-r!r;rgr)rrr)rrr-r!rgrr;s r dropdown_staterGs34GHHUIMFE"^&:L_H_&`&`&``N ;;N;;;;;'J6SX'Y'Y$uuuP^wI J JJrT)visibler elem_classes update_formrrrc t|r|nd|}5|d|d|dd|}|Ft|d|\} } ||t |d| | |d |d |o |ddkd d |} dddn #1swxYwY|dd|} fd} fd}|sH| | g}|jg| || g|d| jg||| g|d| || dS)Nr0rrgr-zTemporal Upsampling)rr;rr!r?rr)titlemarkdownrr!r Multiplierr;F)r;rrczt||\}}}||t||fS)N)rr;r)rupdater})r-r;rr!grs r refresh_methodz5create_generation_temporal_ui..refresh_methodZsA+YZ`bg+h+h(E5yy!35$v,,yWWY^^^rcxt||\}}}|t||fS)N)r;r)rrr})r-r!rr;rs r refresh_multiplierz9create_generation_temporal_ui..refresh_multiplier^s9=feLL5%yyud6lly;;UBBrhidden)triggersfninputsoutputs show_progress)r;r- multiplier) rRowDropdownrbindrTextboxoninput)rrrrrrrstater- help_title help_markdownrr;rrr s` r create_generation_temporal_uirPsU 'A..rWj k k kE U+;%>@@ r%&9::GN8KNNNNw-..a**+>P_*```BQ Y > >WYYr)main_offloadobjloaded_model_contextc t|}|;t|pdrtd|d|||fS|d}t |dr|nd} t|||} t5 ||j n|} | | | %t |dr| n | t |dr|j |fi||j ||||fd | i|| | j n-| rtj|gntj|gcdddS#| | j w| rtj|gwtj|gwxYw#1swxYwYdS) Nr0rrrpersistent_modelsFrelease_private_runtimeload_upsamplerr )rrr1r:rqror"rr offloadobj unload_allr#r$temporal_upsampler unload_vram release_all) rsampleprevious_last_framefpsrr kwargsrr persistentborrowed_contextcore_offloadobjs r r'r'us%&9::G "(b ) ) / / 1 1 ^\FY\\\]] ]*C//  / / 1 1& 9D07AT0U0U`**,,,[`J.w8KMabb  ! !55 5AUAa2==gvO***,,,+AZ0[0[+//1111!)gg?O.P.P)&&':EEfEEE,7,-@&J]_bUUzJUNTUU+ +668888 5 ,dV4444 ,dV444!55555555 + +668888 5 ,dV4444 ,dV4444!5555555555s-,G .A:E>(A G >A G  G  GGc ht|}|t|dsdS|j|fd|i|S)NdownloadFr)rror2)r process_filesr-rs r download_for_valuer4sJ%&9::Gggz::u 7 M ] ]?R ]V\ ] ]]rcg}t|D]y}t|dr*||| 344  !>HHH Nrcht|dr!t|niS)Nrm)rorprmrHs r _default_configrCs2-4W>N-O-O W4&&(( ) ) )UWWrrtct|dr"t||nt|S)Nnormalize_config_section)rorprE)rrts r _normalize_config_sectionrFs==DWNh=i=i {40088 9 9 9ostzo{o{{rc |ti}t|tr#|t |ini}it |t|tr|ni}t ||SrF)r]r?rrpr=rCrF)rxrr@nestedvaluess r read_config_sectionrJs =rBBGAKGUYAZAZ bW[[/88" = = =`bF Y(( Yz&$7O7O-WVVUW YF $Wf 5 55rct|it|t|pi}|t|t |<|SrF)rFrCrprAr=)rxrrts r write_config_sectionrLsS &w0b?73K3K0btTZT`^`OaOa0b c cFFLOM""#9'#B#BC Mrc t|td}tD]0}t |rt ||t ||1|t|tdkSrF)reprr]r?rrCrLrJ)rxbeforers r !migrate_temporal_upsampler_configrPs -##$A4HH I IF/ff 7 # # f 9L]\c9d9d e e e T-++,I4PPQQ QQr lock_configrR$list[TemporalUpsamplerConfigBinding]c .g}tD]}t|dst||t||}||||}|r>|t |t|t||S)Ncreate_config_uirQ) rrorLrJrUrGrr=list)rrxrRbindingsrrtrs r rUrUsH/vvw 233  %mW>QR_ah>i>ijj++BK+PP  v OO:7DZ[bDcDceijresestt u u u OrrWcd|DS)Nc,g|]}|jD]\}}|Sr)r)rZbindingr components r r[z%config_components..s, S S S''BR S S,!YI S S S Srr)rWs r config_componentsr\s S SX S S SSrcft|pg}id}}|D]v}i}|jD]7\}}|t|krtd||||<|dz }8t |jit |j|||j<w|t|krtd|S)NrzDTemporal upsampler config UI values do not match registered controlsr)rVrr9r:rFrrCr)rWrIupdatesindexrZrtfieldrs r collect_config_updateras &,B  FUGAA(  HE1F ## !ghhh"5MF5M QJEE&?QTcdkdsTtTtQx~Q'A'A"## F _``` Nrr^ list[str]c<g}|D]}|j|vrt|jdrv|j||j}t |t r|r||rt |tr|d|Dz }|S)Nvalidate_config_sectioncg|]}||Srr)rZr<s r r[z3validate_config_update_messages..s>>>d>T>>>r)rrorrdrrrGrV)rWr^messagesrZmessages r validate_config_update_messagesrhsH??   ( (WW_F_-`-` (o==ggFX>YZZG'3'' ?G ?((((GT** ?>>g>>>> Orcd|D],}|j|vr!t||j||j-dSrF)rrLr)rxrWr^rZs r apply_config_updaterjsJ^^   ( ( I[A\ ] ] ]^^r old_config new_configc"t|}tD]w}t|dst||}t||}t|dr||||}n||k}|r|xdS)N release_vramconfig_requires_release)setrrorJrorn)rkrl changed_keysr old_section new_sectionshould_releases r *release_changed_config_temporal_upsamplersrus|$$L/ # #w//  )*g>> )*g>> 75 6 6 8$<<[+WcddNN(K7N  #  " " " # #rc,eZdZdZdZdZdZd dZdS) SimpleScaleSuffixMixinz[Value helpers writing '*' while accepting the legacy concatenated form.cXdt|DS)Ncg|]\}}|Srrrs r r[z7SimpleScaleSuffixMixin._method_keys..sWWW3WWWr)r_rq)selfs r _method_keysz#SimpleScaleSuffixMixin._method_keyss*WW/$2S2S2U2U"V"VWWWWrc(t|pd}t|t dD]2}||rt||d}|dn||fcS3dS)Nr0T)rreverser)rr1r2rr{r9r8r>)rzr;r<r-r!s r rz"SimpleScaleSuffixMixin.split_values5;B%%''--//T..00c4HHH B BFv&& B/fcBB$}tt65/AAA Btrct|pd}||vrdS|di|d}t |pd}||vr$t||pd}t||S)Nr0rrr) rr1r2r{rqr]r"rr4)rzr-r!rs r rz"SimpleScaleSuffixMixin.build_valuesV\r""((**0022 **,, , ,47799==mRPPTTU[]_`` ejq!!  # #01R1R1T1TV\]]babE&vu555rr#r}c0||duSrF)r)rzr;s r rz$SimpleScaleSuffixMixin.is_upsampling)s&&d22rNr#r})rrr__doc__r{rrrrrr rwrws^eeXXX666333333rrw)r!r"r#r)r-rr!r"r#r)r-rr5r"r#r6)r?rr!r"r#r)r#rD)rJr)rTrUr#rV)rTrUr#r`)rTrUrerr#rrF)rirjr#rk)rirjr#rD)F)r|r}r#rr)r|r}r#r)rTrUr#r")rTrUr-rr#r")r#r)r#r)r#r)r#r)rTrUr-rr#r6)r#r")r#r)r#r)r#r)r#r)r|r}r#rV)r#r)rr}r#rU)rr}rr}rr}r#rU)rr}r#r)T)rxrUr#rU)r#rU)rtrUr#rU)rxrUrtrUr#rU)rxrUr#r})rxrUrRr}r#rS)rWrSr#r)rWrSr#rk)rWrSr^rkr#rb)rxrUrWrSr^rkr#rD)rkrUrlrUr#rD)Tr __future__r dataclassesrrrOtypingrshared.attentionr shared.utilsrprocessor_statusr r r r r model_contextrr?r3rnrrrprrr)r,r4r>rCrIrSr_rdrhrur{rrrrrrrrrrrrrrrrrrrrrrrrrrr'r4r6r=rArCrFrJrLrPrUr\rarhrjrurwrrr rs.#"""""!!!!!!333333))))))UUUUUUUUUUUUUU222222!6CD+-,,,,9<,>>>> $$$$$$$ $ CCCC ****bbbb$]]]]5555 EEEE QQQQMMMM!!!!     ?????qqqqq@@@@ppppp%%%%{{{{ eeee]]]]    rrrrTTTTCCCC KKKK     F F F F[[[[[[[[ KKKKK:>!!!!!!.7;______]]]] HMJJJJJJOSpuEI_dquHHHHHH2RW Z Z Z Z Z Zae{555558^^^     QQQQ XXXX||||6666 RRRRPU      TTTT        ^^^^ # # # #3333333333r