oFj!5 <ddlZddlZddlZddlZddlmZdZgdZgdZdddd d d d d ddd Z ddddddddddd Z dddZ da dZ dZefdZeefdZd Zd!Zd"Zd#Zd$Zd%Zd&Zd8d(Zd)Zd8d*Zd+Zd,Zd-Zd.Zd/Z d0Z!d1Z"d2Z#d9d3Z$d9d4Z%d5Z&d6Z'd7Z(dS):N)match_nvidia_architecturezresolutions.json))z4096x4096 (1:1) 4096x4096)z5440x3072 (16:9) 5440x3072)z3072x5440 (9:16) 3072x5440)z3840x2176 (16:9) 3840x2176)z2176x3840 (9:16) 2176x3840)z2880x2880 (1:1) 2880x2880)z3840x1664 (21:9) 3840x1664)z1664x3840 (9:21) 1664x3840)z2048x2048 (1:1) 2048x2048)z1920x1920 (1:1) 1920x1920)z2560x1440 (16:9) 2560x1440)z1440x2560 (9:16) 1440x2560)z1920x1440 (4:3) 1920x1440)z1440x1920 (3:4) 1440x1920)z2160x1440 (3:2) 2160x1440)z1440x2160 (2:3) 1440x2160)z1456x1456 (1:1) 1456x1456)z2688x1152 (21:9) 2688x1152)z1152x2688 (9:21) 1152x2688)#)z1920x1088 (16:9) 1920x1088)z1088x1920 (9:16) 1088x1920)z1440x1440 (1:1) 1440x1440)z1536x1024 (3:2) 1536x1024)z1024x1536 (2:3) 1024x1536)z1920x832 (21:9)1920x832)z832x1920 (9:21)832x1920)z2048x768 (8:3)2048x768)z1024x1792 (4:7) 1024x1792)z1088x1088 (1:1) 1088x1088)z1024x1024 (1:1) 1024x1024)z1280x720 (16:9)1280x720)z720x1280 (9:16)720x1280)z1600x400 (4:1)1600x400)z1280x544 (21:9)1280x544)z544x1280 (9:21)544x1280)z1104x832 (4:3)1104x832)z832x1104 (3:4)832x1104)z 960x960 (1:1)960x960)z960x544 (16:9)960x544)z544x960 (9:16)544x960)z 832x624 (4:3)832x624)z 624x832 (3:4)624x832)z 720x720 (1:1)720x720)z832x480 (16:9)832x480)z480x832 (9:16)480x832)z672x384 (16:9)672x384)z384x672 (9:16)384x672)z 512x512 (1:1)512x512)z576x320 (16:9)576x320)z320x576 (9:16)320x576)z 448x448 (1:1)448x448)z 448x256 (7:4)448x256)z 256x448 (4:7)256x448)z 320x320 (1:1)320x320iiiiiiii@8ii) 256p320p384p480p540p720p1080p1440p2160p4096pi@iiiii8iipi)2k4kct|to:tjd|duS)N\d+x\d+) isinstancestrre fullmatchstriplowervalues +C:\Users\johnn\Wan2GP\shared\resolutions.pyis_resolution_valuerRss= eS ! ! abl:u{{}}?R?R?T?T&U&U]a&aac|dd\}}t|t|fS)Nx)rNsplitint)rPwidthheights rQparse_resolutionr[ws:KKMM''Q//ME6 u::s6{{ ""rSc"|dSt|ts|d|ddSg}|D]}t|ttfrIt|dks6t|dtrt|dts|d|d|dSt |ds|d|d|ddS||d|df|S)N"z6" should be a list of 2 elements lists ["Label","WxH"]rrVz," contains an invalid list of two elements: z?" contains a resolution value that is not in the format "WxH": )rIlisttuplelenrJrRappendrN)resolution_choices source_nameprinter normalizedtups rQnormalize_resolution_choicesrh|sD!t ($ / /WKWWWXXXtJ!44#e}-- SQjQTUVQWY\F]F]eopstupvx{e|e| GV VVQTVV W W W44"3q6**  Gl lldghidjll m m m443q63q6<<>>23333 rScpttStj|sgS t |dd5}t j|}dddn #1swxYwYn*#t$r}|d|d|d}Yd}~nd}~wwxYwt|||}|gS|atS)Nrzutf-8)encodingz Invalid "z" : ) _custom_resolutionsospathisfileopenjsonload Exceptionrh)resolution_filerefrcerfs rQload_custom_resolution_choicesrws%&"" 7>>/ * * " /3 9 9 9 .Q!%1  . . . . . . . . . . . . . . . """4O4444555!"..@/SZ[[J $ s;A0A$ A0$A((A0+A(,A00 B:BBc dadSN)rlrSrQreset_custom_resolution_cacher{srSct}g}|D]6\}}||vr |||||f7|Sry)setaddrb)rcseendedupedlabel resolutions rQdedupe_resolution_choicesrsc 55DG/,,z     z*++++ NrScTt|}|dkrtd||S)NrVzInvalid resolution block size: )rX ValueError) block_sizes rQnormalize_block_sizers3ZJA~~G:GGHHH rSc>|dkr|St|||z|zSNrV)max)rPrs rQalign_dimension_to_blockrs*Q z5J.; < <z+align_resolution_choices..s^%% E:#9*j#Q#Q  z3E F FHZ[%%%%%%rS)rr)rcrs `rQalign_resolution_choicesrsW%j11JQ!! $%%%%!3%%%  rSFc\t|rtngttzSry)r_DEFAULT_RESOLUTION_CHOICES_4KDEFAULT_RESOLUTION_CHOICES include_4ks rQbuiltin_resolution_choicesrs) C-- D DtLfGgGg ggrScZttdtzS)NTrrrrwrzrSrQall_global_resolution_choicesrs( $%?4%P%P%PSqSsSs%s t ttrScZtt|tzS)Nrr)enable_4k_resolutionss rQ!default_global_resolution_choicesrsA $%?K`%a%a%aeCeEeE&E F FFrSct|tr(|dvS|dvS)N>0noofffalse)Fr)rIrJrMcasefoldrOs rQ'keep_resolution_on_model_switch_enabledrs@%K{{}}%%''/JJJ  ""rSct|\}}||z}tD]\}}||kr|cStt tSry)r[GROUP_THRESHOLDSitemsnextreversed)resolution_strrYrZ pixel_countgroup thresholds rQcategorize_resolutionrsl$^44ME6&.K,2244y ) # #LLL $ )** + ++rScZt|}|tvr t|S|dr |dd}|sdSt |}|tvr|ndS)Np) rJrMrNCATEGORY_ALIASESendswithisdigitrX GROUP_TIERSvalues)categorykeytiers rQ_category_tierrs h--     % % ' 'C $$ ||C#2#h ;;==t s88D;--////44T9rScd}tjd|t|S)Nct|d}|t|n|dS)Nr)rrrJ)matchrs rQ replace_tokenz5_normalize_category_expression..replace_tokens8ekk!nn-- ,s4yyy%++a..@rSz [a-z0-9]+)rKrrJrMrN) expressionrs rQ_normalize_category_expressionrsHAAA 6, s:/D/D/F/F/L/L/N/N O OOrSc|gSt|tr|gSt|ttfr d|DSgS)Nc,g|]}t|Srz)rJ)rrs rQ z2normalize_category_expressions..sGGGJJGGGrS)rIrJr_r`)category_expressionss rQnormalize_category_expressionsr s[# &,,&$%%&u 66HGG2FGGGG IrSczt|}|sdSt|tfd|DS)NTc3VK|]#}tt|diV$dS)TN)rr)rrrs rQrz#category_allowed..sWBBgq(*H*T*TVZ)[]abbBBBBBBrS)rrany)rr expressionsrs @rQcategory_allowedrsn01EFFK t x D BBBBvABBB B BBrSc fd|DS)NcZg|]'}tt|d%|(SrV)rr)rrrs rQrz;filter_resolution_choices_by_categories..sG     1*Q-@@BV W W    rSrz)rcrs `rQ'filter_resolution_choices_by_categoriesrs.    ,   rSc |s|St|pds|ddSt|\}}||z ||z t| tt  i}|D]>}|t|dg|?t| fd} fd}t|||dS)NrrVct|z tt|tz fSry)absindexr)r group_order target_grouptarget_group_indexs rQz$closest_resolution..3slC @Q@QRW@X@X[m@m.score5s^(33 v$(EFNl#BCCDD 56>]:;;mK K''rS) rRr[rr_rr setdefaultrbmin) target_resolutionrc target_width target_heightgrouped_choicesr closest_grouprrrrrrs @@@@@rQclosest_resolutionr$sW !  06B 7 7(!!$Q''"23D"E"EL--/L =0M():;;L'((K$**<88O$XX""#8#C#CRHHOOPVWWWW.o.o.o.o.o.opppM(((((( }-5 9 9 9! <.LsI9J9Jhr9RV`abVc9c9J9J9J9J9J9JrSrrV) getrhrrrrrrr)r model_defrrmodel_resolutionsrcs` rQresolve_resolution_choicesr>s! mT::$9:KM`aageg ==14 8 8 D!:;MPwyVyXyXZcd|Z}Q~Q~<~""  / 6 6 BDEbEdEdfoqIgJKK>?TUU12DmwmimmTdfhFiFiFiFPQQ 4x ,S9J9J9J9JwI9J9J9J6J6J,$67PRd$e$e!! " */A *$6q$9!$<! 8 88rScXt|pdsdSt||||dS)NrrV)rRr)source_resolutiontarget_model_defrrs rQresolve_model_switch_resolutionrTs; 06B 7 7t %&79IK`bl m mno pprScpi|D]@}t|d}|g|AsggdfSfdtD}|| |r |d}nt|}|vr |r|d}||g|fS)NrVcg|]}|v| Srzrz)rrgrouped_resolutionss rQrz,group_resolution_choices..cs$\\\%uH[?[?[?[?[?[rSr)rrrbrreverser)rcselected_resolutionrravailable_groupsselected_grouprs @rQgroup_resolution_choicesr Zs(EE %jm44&&ub1188DDDD 2t|\\\\+;\\\"'7")!,./BCC !4 4 49I 4-a0N 044^RHH. XXrSc fd|DS)NcFg|]}t|dk|Srr)rrr s rQrz!group_choices..qs4 v v v:?TU_`aUb?c?cgu?u?uJ?u?u?urSrz)rcr s `rQ group_choicesrps v v v v); v v vvrSc*||t|<|Sryr)last_resolution_per_grouprs rQremember_last_resolutionrtsCM3J??@ $$rS)F)FN))rqrrmrKshared.match_archirRESOLUTION_FILErrrrrrlrRr[printrhrwr{rrrrrrrrrrrrrrrrrrrr rrrzrSrQrs 888888%!!!2+++\                            bbb### KP&4CE( === kkk WWWhhhhuuuFFFF### ,,, : : :PPPBBB===49999,qqqq YYY,www%%%%%rS