RЦi8t%SrSSKrSSKrSSKrSSKrSSKJrJr SSKJ r SSK J r SSK J r JrJrJrJrJrJrJrJrJr SSKJrJr /S Qr\"\5r\\\\4\S '0r\\\4\S '0r \\\S \ 44\S '\"5r!\\\S'0r"\\\4\S'0r#\\\4\S'\"\$5r%\\\\4\S'\"\&5r'\\\\\\44\S'0r(\\\\4\S'SS&\\\\4S(\\\\4S)\2S*\\\\4S+\2S,\\2S\\4S-jjr3S?S&\\\\4S*\S\\4S.jjr4Smodelr1is_default_setpriority default_cfgs r2rr+s j !C%K  a  ""Amm )!, -+GZc1B1Y>GYiinj     ' ' , OOE " !:!:    ' ' ,    # #C ( (,K % !/2 Jr3fnc l[RURnURRS5n[ U5(aUSOSnUR n[ US5(aURRU5 OU/UlU[;a)[R"SUSURSUS3SS 9 U[U'U[U'[URU5 [ US 5(GaXAR;GaURUn[!U["5(d8[!U[$5(de['S0UD6n[#[)S/5SU0S 9n[+UR,5GH%upxUS :Hn UR.UnU(aSR1XH/5OUn [%XH(aUOSS 9n UR2(a"UR2S:XaUR2U -U S'[5U40U D6nU (a/U[6U'UR8(a[:RU5 U(aJU[6U 'UR8(a[:RU 5 [<URU 5 GM [<URU5 GM( U[>U'U$)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+A#B'rKJsI :&!l '' :,&8q U` ` &(z"#.Z k"&&z2sN## 6F6F(F&&z2 +z22k400 00*9[9N$%+RTc STS35eT(aSRTRT/5O TRn[R"STSUS3SS9 UR SS5nT"S X=(d TS .UD6$) NzModel z& has been removed with no replacement.r.zMapping deprecated model name z to current rSrTrf) pretrainedrfr9)rbr]r_r`pop)rpkwargs current_namerfrlrmrks r2_fn#_deprecated_model_shim.._fns%h&0AAg(hh%GRsxx!4!4k BCXbXkXk  66G|T`Saabcpqr$4d;hZ@]R]haghhr3)Fr9)rkrlrmrts``` r2_deprecated_model_shimrvsii Jr3redeprecation_mapc[RUnURS5n[U5(aUSOSnUR 5HupE[ US5(aUR RU5 SnSnU(a[U5up[X(5n[XFU5n [X$U 5 U [U'U[U'[URU5 U[ U'U["UU'M g)Nr.rPrQrR)rYrZr/r\r;r^rRrErgetattrrvsetattrr#r"r!rCr)r() rerwrcrd deprecatedcurrentrlrmrsdeprecated_entrypoint_fns r2register_model_deprecationsr~s ++k "C#))#.+./@+A+A#B'rK.446 3 " " KK  z *  (24((8W]]_2U V2UQiikkCFq (2U VV Vs+AfiltercZ[U5upU(dSRUS/5U/$U/$)z;expand a 'base_filter' to 'base_filter.*' if no tag portionr.r8)rrb)r filter_base filter_tags r2_expand_filterrs226:K +s+,f55xr3modulerpexclude_filtersname_matches_cfg include_tagscU(a![U[[45(aUOU/nO/nUcUnU(d[[R 55nOa[U[ 5(a [UnOB[U[5(de[5nUHnUR[U5 M U[R 5- nU(ay[5n UHnU R[U5 M U nUV V s/sHn [U 5HoPM M nn n UV V s/sHn [U 5HoPM M nn n U(aM[5n UH<n [R"Xz5n [U 5(dM+U R!U 5n M> OUn U(a`[U[[45(dU/nUH<n[R"X5n[U5(dM+U R#U5n M> U(a[$R'U 5n U(a[[(5R'U 5n [+U [,S9$s sn n fs sn n f)alReturn list of available model names, sorted alphabetically Args: filter - Wildcard filter string that works with fnmatch module - Limit model selection to a specific submodule (ie 'vision_transformer') pretrained - Include only models with valid pretrained weights if True exclude_filters - Wildcard filters to exclude models after including them with filter name_matches_cfg - Include only models w/ model_name matching default_cfg name (excludes some aliases) include_tags - Include pretrained tags in model names (model.tag). If None, defaults set to True when pretrained=True else False (default: None) Returns: models - The sorted list of models Example: model_list('gluon_resnet*') -- returns all models starting with 'gluon_resnet' model_list('*resnext*, 'resnet') -- returns all models with 'resnext' in 'resnet' module )key)r<tuplelistr:r#keysstrr!rupdater)r'rfnmatchrr\union differencer$ intersectionr&sortedr)rrrprrrinclude_filters all_modelsmmodels_with_tagsfefmodelsinclude_modelsxfexclude_modelss r2rrs4$.vt}$E$E&F8! "#5#:#:#<= fc " "#4V#""n5! /E4=99./O!B$^^F7N>""**>:" &33F;+,99&A &l ++5TSs I,Ic[USUSS9$)NT)rrprr)r)rrs r2rr s '  r3cDU(a [UO[n[U5$N)r(r)r)rall_deprecateds r2get_deprecated_modelsrs=C1&9I[N N ##r3c*[U5nU[;$)z!Check if a model name exists )rr#)r* arch_names r2rrsj)I * **r3 module_filterc[U5nU(a,U[RU05;a[SUSUS35e[U$)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@[R5n[U5$)zHReturn list of module names that contain models / model entrypoints )r!rr)rZs r2rr-s $$&G '?r3 module_namesc^[U5m[U[[[45(de[ U4SjU55$)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 c3<># UHnT[U;v M g7fr)r!).0nrs r2 &is_model_in_modules..?sG,Qy-a00,s)rr<rrr:any)r*rrs @r2rr4s:j)I lUD#$6 7 77 7 G,G GGr3cU[;$r)r$r5s r2rrBs . ..r3allow_unregisteredcU[;a[[U5$[U5up#U[;a[ SUSUS35eU(ag[ SUS35e)NzInvalid pretrained tag (z) for r.zModel architecture (z#) has no pretrained cfg registered.)r&rrr%r)r*rrr1s r2get_pretrained_cfgrFsh++.z:;;)*5NI''5cU& 1MNN -i[8[\ ]]r3cfg_keyc.[USS9n[X!S5$)zNGet a specific model default_cfg value by key. None if key doesn't exist. F)rN)rry)r*rcfgs r2rrSs ZE BC 3 &&r3cn[U5up[UnUVs0sH oD[U_M nnU$s snf)z6Get all pretrained cfgs for a given architecture. )rr'r&)r*r_ model_namesrr6s r2r r ZsA( 3LI"9-K2= >+Q%a( (+D > K ?s2)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)rrrrrvr~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