a
    dh                     @   s0   d dl Z d dlmZmZ ee jjdddZdS )    N)
OpOverloadOpOverloadPacketopgraphc                 C   s:   t | trJ d|  t | ts&J tj| j| d S )Nz;Must pass specific op overload, not overload packet, found )
isinstancer   r   torch_CZ&_jit_register_decomposition_for_schemaZ_schemar    r
   g/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/jit/_decomposition_utils.py_register_decomposition   s    r   )r   Z
torch._opsr   r   r	   ZGraphr   r
   r
   r
   r   <module>   s   