a  þd³ã@s2ddlZddlTe e¡Zd dd„Zd dd„ZdS) éNé)Ú*FcCs:|s|s|sJdƒ‚|pi}|p"i}|s@|dur@t|dƒr@|j}i}d}| dd¡dur`|d}|ddf}| dd¡durºt|dttfƒsJ‚t|dƒdks¤J‚t|dƒ}|d}nr| dd¡durðt|dtƒsÜJ‚||d|df}n<|r| d d¡dur|d }n| dd¡dur,|d}||d<d |d <| d d¡rX|d |d <n| d d¡rr|d |d <t|d <| d d¡durÔt|d ƒ}t|ƒd kr¸tt|ƒ|ƒ}nt|ƒ|ksÊJ‚||d <n| d d¡rî|d |d <t |d<| dd¡durPt|dƒ} t| ƒd kr4tt| ƒ|ƒ} nt| ƒ|ksFJ‚| |d<n| dd¡rj|d|d<t } | dd¡r†|d} n4|r¤| dd¡r¤|d} n| dd¡rº|d} | |d<t } | dd¡rÞ|d} n| dd¡rô|d} | |d<|r6t   d¡| ¡D] \} } t   d| t| ƒf¡q|S)NzHAt least one of model, args, or pretrained_cfg required for data config.Úpretrained_cfgéZchanséàÚ input_sizerZimg_sizeZtest_input_sizeZbicubicÚ interpolationÚmeanrÚstdÚcrop_pctZ test_crop_pctÚ crop_modez:Data processing configuration for current model + dataset:z %s: %s)ÚhasattrrÚgetÚ isinstanceÚtupleÚlistÚlenÚintZIMAGENET_DEFAULT_MEANZIMAGENET_DEFAULT_STDZDEFAULT_CROP_PCTZDEFAULT_CROP_MODEÚ_loggerÚinfoÚitemsÚstr)ÚargsrÚmodelÚ use_test_sizeÚverboseZ data_configZin_chansrr r r r ÚnÚv©rúY/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/timm/data/config.pyÚresolve_data_configs‚               r cCst|||||dS)ab Resolve Model Data Config This is equivalent to resolve_data_config() but with arguments re-ordered to put model first. Args: model (nn.Module): the model instance args (dict): command line arguments / configuration in dict form (overrides pretrained_cfg) pretrained_cfg (dict): pretrained model config (overrides pretrained_cfg attached to model) use_test_size (bool): use the test time input resolution (if one exists) instead of default train resolution verbose (bool): enable extra logging of resolved values Returns: dictionary of config )rrrrr)r )rrrrrrrrÚresolve_model_data_configesûr!)NNNFF)NNFF)ÚloggingÚ constantsÚ getLoggerÚ__name__rr r!rrrrÚs û _û