^j8@UdZddlZddlZddlZddlZddlmZmZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZmZmZmZddlmZmZgd ZeeZeeeefed <iZeeefed <iZ eeed e ffed <eZ!eeed<iZ"eeefed<iZ#eeefed<ee$Z%eeeefed<ee&Z'eeeeeeffed<iZ(eeeefed<dd&eeeefd(eeeefd)e2d*eeeefd+e2d,ee2deefd-Z3 d?d&eeeefd*edeefd.Z4dmodelr1is_default_setpriority default_cfgs r2rr+s j !C%K "1 a ""Amm )!, s+-+GZc1B1Y>GYiin%j     ' ' , OOE " !:!:    ' ' ,    # #C ( (,K % ! /"2 Jr3fnc tj|j}|jjd}t |r|dnd}|j }t |dr|jj|n|g|_|tvr+tjd|d|jd|dd |t|<|t|<t|j|t |d r||jvr{|j|}t!|t"s6t!|t$sJt'di|}t#t)dgd|i }t+|j,D]\}}|d k(} |j.|}|rdj1||gn|} t%||r|nd } |j2r!|j2dk(r|j2| z| d<t5|fi| }| r*|t6|<|j8rt:j||rC|t6| <|j8rt:j| t<|j| t<|j||t>|<|S)Nr.__all__z Overwriting z in registry with zr. This is because the name being registered conflicts with an existing name. Please check if this is not expected. stacklevel default_cfgs)rAr6r) architecturer1ztimm/ hf_hub_idr9) sysmodules __module__r/len__name__hasattrrRrEr#warningswarnr"r!rCrVr<rr=rr enumeraterAr6joinrXrr&r>r$r'r%) rNmodmodule_name_split module_namer*rMpretrained_cfgtag_idxr1 is_defaultmodel_name_tag replace_itemss r2rrKs[ ++bmm $C ++C0+./@+A#B'rKJsI :&!l '' :,&8q U` ` &(z"#.Z k"&&z2sN# c6F6F(F&&z2 +z2k40 00*9[9N$%+R._fns%h&0AAg(hh%GRsxx!4!4k BCXbXkXk  66G|T`Saabcpqr$4d;hZ@]R]haghhr3)Fr9)rkrlrmrts``` r2_deprecated_model_shimrusi Jr3redeprecation_mapctj|}|jd}t|r|dnd}|j D]\}}t |dr|j j|d}d}|rt|\}}t||}t|||} t||| | t|<|t|<t|j||t |<|t"||<y)Nr.rPrQrR)rYrZr/r\r;r^rRrErgetattrrusetattrr#r"r!rCr)r() rervrcrd deprecatedcurrentrlrmrsdeprecated_entrypoint_fns r2register_model_deprecationsr}s ++k "C#))#.+./@+A#B'rK.446H G 3 " KK  z *  ("n5 6 /E4=9./O! ;B$^^FB7N>"**>: ; &33F;+,99&A &l ++5TSs #HH c t|d|dS)NT)rrprr)r)rrs r2rr s '  r3c:|r t|nt}t|SN)r(r)r)rall_deprecateds r2get_deprecated_modelsrs=C1&9I[N N ##r3c(t|}|tvS)z" Check if a model name exists )rr#)r* arch_names r2rrsj)I * **r3 module_filterct|}|r*|tj|ivrtd|d|dt|S)z6Fetch a model entrypoint for specified model name zModel (z not found in module r.)rr!get RuntimeErrorr#)r*rrs r2rr$sMj)I*;*?*? r*RRWZL0Em_TUVWW i ((r3c@tj}t|S)zI Return list of module names that contain models / model entrypoints )r!rr)rZs r2rr-s $$&G '?r3 module_namesc|t|t|tttfsJt fd|DS)zCheck if a model exists within a subset of modules Args: model_name - name of model to check module_names - names of modules to search in c34K|]}t|vywr)r!).0nrs r2 z&is_model_in_modules..?sGQy-a00Gs)rr<rrr:any)r*rrs @r2rr4s7j)I lUD#$6 77 7 G,G GGr3c|tvSr)r$r5s r2rrBs . ..r3allow_unregisteredc|tvrtt|St|\}}|tvrt d|d|d|ryt d|d)NzInvalid pretrained tag (z) for r.zModel architecture (z#) has no pretrained cfg registered.)r&rrr%r)r*rrr1s r2get_pretrained_cfgrFsj++.z:;;)*5NIs''5cU& 1MNN -i[8[\ ]]r3cfg_keyc6t|d}t||dS)zO Get a specific model default_cfg value by key. None if key doesn't exist. F)rN)rrx)r*rcfgs r2rrSs ZE BC 3 &&r3clt|\}}t|}|Dcic] }|t|}}|Scc}w)z7 Get all pretrained cfgs for a given architecture. )rr'r&)r*r_ model_namesrr6s r2r r ZsD( 3LIq"9-K2= >QA%a( ( >D > K ?s1)rQ)NrQ)rQrQFrQFN)rQrQr)T)>__doc__rrrYr_ collectionsrrcopyr dataclassesrtypingrrr r r r r rrr _pretrainedrrrRr:r!r__annotations__r"r#r$r%r&rr'r=r(r)rrrrrur}rrrboolrrrrrrrrrrr r9r3r2rs *]]]2  *5S)94SX &9#%$sCx.%46DhsCx0016"%%s3x'02T#},-235S-/05)4T):$sDI~&:DOPTDUd3S(3--?(@#@AU/1Dhsm+,1S#uS#X /c/c/S%S#X 0M*N%N O@=xS)=hsCx.@=@CX[^HSH4XVY]HZC[H,W#W$uS#X"7W 3)+(* 13!&'+ P,c49n%P,c49n%P,P,sDI~. P,  P, tn P, #YP,h)+! c49n%   #Y $#$tCH~$ +++))Xc])hWZ\_W_N`)d3i H H',U38_d3iS-Q'R H  H/C/D/ ^3 ^D ^HUbLc ^''s'x}'c=6H1Ir3