o
    h                     @   s   U d dl mZmZ z0d dlmZmZmZmZmZ dZ	ddedddiiZ
eeeeeef ef f ed< ee
 W n eyW   d	d
 ZG dd dZdd Zdd ZdZ	Y nw dd Zg dZdS )    )DictUnion)DeviceLayerRepositoryregister_kernel_mappingreplace_kernel_forward_from_hubuse_kernel_forward_from_hubTMultiScaleDeformableAttentioncudaz!kernels-community/deformable-detr)repo_id
layer_name_KERNEL_MAPPINGc                  O   s   dd }|S )Nc                 S   s   | S N )clsr   r   y/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/transformers/integrations/hub_kernels.py	decorator+      z.use_kernel_forward_from_hub.<locals>.decoratorr   )argskwargsr   r   r   r   r   *   s   r   c                   @   s   e Zd Zdd ZdS )r   c                 O      t d)NzNLayerRepository requires `kernels` to be installed. Run `pip install kernels`.RuntimeError)selfr   r   r   r   r   __init__1      zLayerRepository.__init__N)__name__
__module____qualname__r   r   r   r   r   r   0   s    r   c                  O   r   )Nz^replace_kernel_forward_from_hub requires `kernels` to be installed. Run `pip install kernels`.r   r   r   r   r   r   r   4   s   r   c                  O   r   )NzVregister_kernel_mapping requires `kernels` to be installed. Run `pip install kernels`.r   r   r   r   r   r   9   r   r   Fc                   C   s   t S r   )_hub_kernels_availabler   r   r   r   is_hub_kernels_available?   r   r!   )r   r!   r   r   r   N)typingr   r   kernelsr   r   r   r   r   r    r   str__annotations__ImportErrorr!   __all__r   r   r   r   <module>   s*   $	