a dW @sddlZddlmZddlmZmZmZdgZdddZedd Z ed d Z ed d Z eddZ ddZ eeeeeedddZeeefeddddZdS)N)repeat)ListDictAny'consume_prefix_in_state_dict_if_presentparsecsfdd}||_|S)Ncs$t|tjjrt|Stt|SN) isinstance collectionsabcIterabletupler)xn_/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/nn/modules/utils.pyr sz_ntuple..parse)__name__)rnamerrrr_ntuples r_single_pair_triple _quadruplecstfddt|DS)zReverse the order of `t` and repeat each element for `n` times. This can be used to translate padding arg used by Conv and Pooling modules to the ones used by `F.pad`. c3s |]}tD] }|VqqdSr)range).0r_rrr z(_reverse_repeat_tuple..)r reversed)trrrr_reverse_repeat_tuplesr%)out_sizedefaultsreturncCsVt|tr|St|t|kr4tdt|dddt||t| dDS)Nz%Input dimension should be at least {}rcSs g|]\}}|dur|n|qSrr)rvdrrr (sz&_list_with_default..)r intlen ValueErrorformatzip)r&r'rrr_list_with_default!s r1) state_dictprefixr(cCst|}|D],}||r|t|d}||||<qd|vr|d}t|D]0}t|dkrlqZ|t|d}||||<qZdS)aStrip the prefix in state_dict in place, if any. ..note:: Given a `state_dict` from a DP/DDP model, a local model can load it by applying `consume_prefix_in_state_dict_if_present(state_dict, "module.")` before calling :meth:`torch.nn.Module.load_state_dict`. Args: state_dict (OrderedDict): a state-dict to be loaded to the model. prefix (str): prefix. N _metadatar)sortedkeys startswithr-poplist)r2r3r6keyZnewkeymetadatarrrr-s   )r)r itertoolsrtypingrrr__all__rrrrrr%r,r1strrrrrrs