a  d@sdZddlZddlZddlmZmZmZmZddlZddl m m Z ddl mZeeZd eeeeeeeeed d d ZdS) zQ Position Embedding Utilities Hacked together by / Copyright 2022 Ross Wightman N)ListTupleOptionalUnion) to_2tuplebicubicTF)new_sizeold_sizenum_prefix_tokens interpolation antialiasverbosec Cs6|jd}|d|d|}||kr:|d|dkr:|S|sXtt||} | | f}|r|ddd|f|dd|df} }n d|} }|jd} |d|d|dddddd}tj||||d}|dddddd| }| durtj | |gdd}tj s2|r2t d|d |d |S) Nrr)sizemoder )ZdimzResized position embedding: z to .)shapeintmathsqrtZreshapeZpermuteFZ interpolatetorchcatZjitZ is_scripting_loggerinfo) Zposembr r r r r rZnum_pos_tokensZnum_new_tokensZhwZ posemb_prefixZ embed_dimr^/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/timm/layers/pos_embed.pyresample_abs_pos_embeds& ,  $ r )NrrTF)__doc__loggingrtypingrrrrrZtorch.nn.functionalnnZ functionalrZhelpersr getLogger__name__rrstrboolr rrrrs(