a  þdVã@s4dZddlmZddlmZddlmZdd„ZdS)zQ Create Conv2d Factory Method Hacked together by / Copyright 2020 Ross Wightman é)Ú MixedConv2d)Ú CondConv2d)Úcreate_conv2d_padcKsÆt|tƒr\d|vsJ‚d|vrF| d¡}||kr:d|d<n |dksFJ‚t|||fi|¤Ž}nf| dd¡}|rp|n | dd¡}d|vrª|ddkrªt|||fd|i|¤Ž}nt|||fd|i|¤Ž}|S)zà Select a 2d convolution implementation based on arguments Creates and returns one of torch.nn.Conv2d, Conv2dSame, MixedConv2d, or CondConv2d. Used extensively by EfficientNet, MobileNetv3 and related networks. Z num_expertsÚgroupsTÚ depthwiserFé)Ú isinstanceÚlistÚpoprrr)Z in_channelsZ out_channelsZ kernel_sizeÚkwargsrÚmr©r úb/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/timm/layers/create_conv2d.pyÚ create_conv2d s      rN)Ú__doc__Z mixed_conv2drZ cond_conv2drZ conv2d_samerrr r r rÚs