
    Ng                        d dl T d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dl	T d dl
T d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dl T d dl!T d dl"T d dl#T d dl$T d dl%T d dl&T d dl'T d dl(T d dl)T d dl*T d dl+T d dl,T d dl-T d dl.T d dl/T d dl0T d dl1T d dl2T d dl3T d dl4T d dl5T d dl6T d dl7T d dl8T d dl9T d dl:T d dl;T d dl<T d dl=T d dl>T d dl?T d dl@T d dlAT d dlBT d dlCT d dlDT d dlET d dlFT d dlGT d dlHT d dlIT d dlJT d dlKT d dlLT d dlMT d dlNT d dlOT d dlPT d dlQT d dlRmSZSmTZTmUZUmVZVmWZWmXZX d dlYmZZZm[Z[m\Z\ d dl]m^Z^m_Z_m`Z`maZambZb d dlcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZm d dlnmoZompZpmqZqmrZrmsZs d dltmuZumvZvmwZw d dlxmyZymzZzm{Z{m|Z|m}Z}m~Z~mZmZ d d	lmZmZmZ d d
lmZ d dlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ dS )   )*)build_model_with_cfgload_pretrainedload_custom_pretrainedresolve_pretrained_cfg set_pretrained_download_progressset_pretrained_check_hash)create_modelparse_model_namesafe_model_name)FeatureInfoFeatureHooksFeatureHookNetFeatureListNetFeatureDictNet)
FeatureGraphNetGraphExtractNetcreate_feature_extractorget_graph_node_namesregister_notrace_moduleis_notrace_moduleget_notrace_modulesregister_notrace_functionis_notrace_functionget_notrace_functions)clean_state_dictload_state_dictload_checkpointremap_state_dictresume_checkpoint)load_model_config_from_hfload_state_dict_from_hfpush_to_hf_hub)model_parametersnamed_applynamed_modulesnamed_modules_with_paramsgroup_modulesgroup_parameterscheckpoint_seqadapt_input_conv)PretrainedCfg
DefaultCfgfilter_pretrained_cfg)adapt_model_from_string)split_model_name_tagget_arch_namegenerate_default_cfgsregister_modelregister_model_deprecationsmodel_entrypointlist_modelslist_pretrainedget_deprecated_modelsis_modellist_modulesis_model_in_modulesis_model_pretrainedget_pretrained_cfgget_pretrained_cfg_valueget_arch_pretrained_cfgsN)beitbyoanetbyobnetcaitcoatconvit	convmixerconvnextcrossvitcspnetdavitdeitdensenetdladpnedgenextefficientformerefficientformer_v2efficientnetefficientvit_mitefficientvit_msraevafastvitfocalnetgcvitghostnethardcorenashgnethierahieradet_sam2hrnetinception_nextinception_resnet_v2inception_v3inception_v4levitmaxxvitmambaout
metaformer	mlp_mixermobilenetv3	mobilevitmvitv2nasnetnestnextvitnfnetpitpnasnetpvt_v2rdnetregnetrepghostrepvitres2netresnestresnetresnetv2rexnetselecslssenet	sequencersknetswin_transformerswin_transformer_v2swin_transformer_v2_crtiny_vittnttresnettwinsvgg	visformervision_transformervision_transformer_hybridvision_transformer_relposvision_transformer_samvitaminvolovovnetxceptionxception_alignedxcit_builderr   r   r   r   r   r	   _factoryr
   r   r   	_featuresr   r   r   r   r   _features_fxr   r   r   r   r   r   r   r   r   r   _helpersr   r   r   r   r    _hubr!   r"   r#   _manipulater$   r%   r&   r'   r(   r)   r*   r+   _pretrainedr,   r-   r.   _pruner/   	_registryr0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?        P/var/www/html/ai-engine/env/lib/python3.11/site-packages/timm/models/__init__.py<module>r      s                                                                       ! ! ! !                                                             " " " "                                                                                                                             " " " " % % % %                         ! ! ! ! ( ( ( ( ( ( ( ( % % % %                        @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ E E E E E E E E E E ` ` ` ` ` ` ` ` ` ` ` ` ` `J J J J J J J J J J J J J J J J J J J J J J J J n m m m m m m m m m m m m m T T T T T T T T T TF F F F F F F F F F F F F F F F F F F F I I I I I I I I I I + + + + + +                                     r   