khUddlZddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZddlmZddlmZddlmZmZmZmZddlmZdd lmZdd lmZm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&dd l'm(Z(dd l)m*Z*ddl+m,Z,ejZe.Z/da0da1e2ejfjidddkDZ5gdZ6edejnZ8de e9efde e9e9ffdZ:d4de;ddfdZ d6dejnde e e9efde2de2d e e d!e;de ee9efddfd"Z?de e9efde e9effd#Z@d$e e9efd%e e9ddfd&ZAd7d'ZB d8d(e9de ee9e e9effd)e e e9efde(fd*ZC d9d+eee8e d,e8ffd(e9d-e;de e d)e e d.e ed/e e d0e;d1e e de ee9efd2e e e9de8fd3ZDy):N)deepcopy)Path) AnyCallableDictListOptionalTupleTypeTypeVarUnion)nn)load_state_dict_from_url)FeatureListNetFeatureDictNetFeatureHookNetFeatureGetterNet)FeatureGraphNet)load_state_dict) has_hf_hubdownload_cached_filecheck_cached_fileload_state_dict_from_hfload_state_dict_from_pathload_custom_from_hf)adapt_input_conv) PretrainedCfg)adapt_model_from_file)get_pretrained_cfgFTIMM_USE_OLD_CACHE) set_pretrained_download_progressset_pretrained_check_hashload_custom_pretrainedload_pretrainedpretrained_cfg_for_featuresresolve_pretrained_cfgbuild_model_with_cfgModelT)boundpretrained_cfgreturnc|jdd}|jdd}|jdd}|jdd}|jdd}d}d}|dk(rtd r d}|sJ|}n[|d k(rd }|}nQ|rd}|}t|ts;J|rd}|}n2d }tr|r t |nd }|s|rtd rd}|}n|rd}|}|dk(r|jd dr||d f}||fS) Nsourceurlfile state_dict hf_hub_idhf-hubT) necessary local-dirFhf_hub_filename)getr isinstancedict_USE_OLD_CACHEr) r* cfg_sourcepretrained_urlpretrained_file pretrained_sdr2 load_frompretrained_locold_cache_valids P/var/www/teggl/fontify/venv/lib/python3.12/site-packages/timm/models/_builder.py_resolve_pretrained_sourcerC+s5##Hb1J#''t4N$((6O"&&|T:M"";5IINX*t"< y" { " ( $I*Nnd3 33 I,N#OGU"3N"C[`"yZ$5O$ !*! !/H!3!34Et!L'8I)JJ n $$enablec|ay)zA Set download progress for pretrained weights on/off (globally). N)_DOWNLOAD_PROGRESSrEs rBr!r![s  rDc|ay)z= Set hash checking for pretrained weights on/off (globally). N) _CHECK_HASHrHs rBr"r"as KrDmodelload_fn cache_dirc|xs t|dd}|stjdyt|\}}|stjdy|dk(rtjdn|dk(rt |t t |}| |||yt|d r|j|ytjd y) a$Loads a custom (read non .pth) weight file Downloads checkpoint file into cache-dir like torch.hub based loaders, but calls a passed in custom load fun, or the `load_pretrained` model member fn. If the object is already present in `model_dir`, it's deserialized and returned. The default value of `model_dir` is ``/checkpoints`` where `hub_dir` is the directory returned by :func:`~torch.hub.get_dir`. Args: model: The instantiated model to load weights into pretrained_cfg: Default pretrained model cfg load_fn: An external standalone fn that loads weights into provided model, otherwise a fn named 'load_pretrained' on the model will be called if it exists cache_dir: Override model checkpoint cache dir for this load r*Nz/Invalid pretrained config, cannot load weights.zHNo pretrained weights exist for this model. Using random initialization.r3zKHugging Face hub not currently supported for custom load pretrained models.r/) check_hashprogressrMr$zXValid function to load pretrained weights is not available, using random initialization.) getattr_loggerwarningrCrrJrGhasattrr$)rKr*rLrMr?r@s rBr#r#gs,$Mwu6F'MN IJ :> JI~ bcHef e - "'  ~& ) * n-rsrD num_classesin_chans filter_fnstrictc`|xs t|dd}|s tdt|\}}|dk(rtj d|} n|dk(rJtj d|d|j d d r|j |yt|} no|d k(rotj d |d|j d d r)t|tt| }|j |y t|dttd|} n|dk(rtj d|dt|ttfrD|j d d } t| t r| dk(rt#g||d|iyt%|d|i} nt%|d|} nt|dk(rNtj d|dt'|} | j)r t+| } n/td||j dd} td| d| || |} |j dd}|S|dk7rNt|t r|f}|D]6}|d z} t-|| || |<tj d!|d"|d#8|j d&d}|j d'd(}|t|t r|f}||d)k7r4|D],}| j3|d zd| j3|d*zd.d }n2|d(kDr-|D](}| |d z}||d| |d z<| |d*z}||d| |d*z<*|j| |+}|j4r2tj d,d-j7|j4d.|j8r3tj1d/d-j7|j8d0yy#t$rt|dtt|} YwxYw#t$r} || } Yd} ~ d} ~ wwxYw#t.$r)} | |=d }tj1d$|d%Yd} ~ d} ~ wwxYw)1a Load pretrained checkpoint Args: model: PyTorch module pretrained_cfg: Configuration for pretrained weights / target dataset num_classes: Number of classes for target model. Will adapt pretrained if different. in_chans: Number of input chans for target model. Will adapt pretrained if different. filter_fn: state_dict filter fn for load (takes state_dict, model as args) strict: Strict load of checkpoint cache_dir: Override model checkpoint cache dir for this load r*NzWInvalid pretrained config, cannot load weights. Use `pretrained=False` for random init.r1z*Loading pretrained weights from state dictr0z&Loading pretrained weights from file () custom_loadFr/z%Loading pretrained weights from url ()rPrOrMcpuT) map_locationrPrO weights_only model_dir)r]rPrOr_r3z2Loading pretrained weights from Hugging Face hub (hfrM)r^rMr5z1Loading pretrained weights from local directory (z#Specified path is not a directory: architecturez this modelz No pretrained weights exist for z). Use `pretrained=False` for random init. first_convz.weightzConverted input conv z pretrained weights from 3 to z channel(s)zUnable to convert pretrained z+ weights, using random init for this layer. classifier label_offsetrrUz.bias)rXzMissing keys (z, zZ) discovered while loading pretrained weights. This is expected if model is being adapted.zUnexpected keys (zY) found while loading pretrained weights. This may be expected if model is being adapted.)rQ RuntimeErrorrCrRinfor7r$rrrGrJr TypeErrorr8listtuplestrrrris_dirrrNotImplementedErrorrSpop missing_keysjoinunexpected_keys)rKr*rUrVrWrXrMr?r@r1r[pretrained_path model_namee input_convsinput_conv_name weight_name classifiersreclassifier_nameclassifier_weightclassifier_bias load_results rBr$r$s($Mwu6F'MN tuu :> JI~L  AC# f  =n=MQOP   mU 3  ! !. 1 (8J e  <^ F ")/M)BK & F ")/M)BK &#F#2i262)E,,E %eW5#+Q}fjjYfhlNm0n ).ZZ A.*$  K '%//-8K+s+)//1 &>>OO$8$?[("0K F*"0K F*"0K D("1K H,!%J"2KH$:;-"HH5(KUL$7  !*  " "< <E1[1:>J!00 LrD)T)NNN)NrrcNTN)r+N)NN)NNNNTNNN)Erloggingoscopyrpathlibrtypingrrrrr r r r r torchr torch.hubrtimm.models._featuresrrrrtimm.models._features_fxrtimm.models._helpersrtimm.models._hubrrrrrrtimm.models._manipulatertimm.models._pretrainedrtimm.models._prunertimm.models._registryr getLogger__name__rRrGrJintenvironr7r:__all__Moduler(rkrCboolr!r"r#r$r%rrr&r'rrDrBrsu SSS.bb40334144 '  H % RZZ^^$8!<=A   +-%tCH~-%%S/-%` T T dd48&*04 .tyy.t c3h0.t(#.tE#t),- .t  .tf48(,04HAyyHA c3h0HAHA HA H% HA  HAE#t),-HA HAVS#X4S>4S>$s)&0V@D;?!! sDcN':!;<!!)c3h 8! !P*.15#'&*"&3704.2wfxV '<<=www! w !) w C= wd^w w'x0wE#t),-w c +w wrD