o
    i                     @   s&   d dl Z d dlZdddZejjZdS )    N   r   circularc                 C   sz   |dkrt j st j rd}d}t| jd D ]}d|| | j|d  ||   ||  f| }qt jjj	| ||dS )Nr   reflect r   r   )mode)
torchjit
is_tracingis_scriptingrangendimshapenn
functionalpad)img
patch_sizepadding_moder   ir   r   2/mnt/c/Users/fbmor/ComfyUI/comfy/ldm/common_dit.pypad_to_patch_size   s   0r   )r   r   )r   comfy.rmsnormcomfyr   rmsnormrms_normr   r   r   r   <module>   s    
