o
    0j                     @   s   U d dl Z d dlmZmZmZmZmZ d dlmZm	Z	 d dl
mZmZ ddlmZ ed Zeed< G d	d
 d
eZG dd deZG dd deZG dd deZG dd deZdS )    N)AnyDictListLiteralOptional)	BaseModelField)	Annotated	TypeAlias   )HPIInfo)ZpaddleZopenvinoZonnxruntimeZtensorrtomInferenceBackendc                   @   s   e Zd ZU eeeddf ed< eed< dZee	 ed< dZ
eed< dZee ed	< dZeeeef  ed
< dZee ed< dZeed< dS )	HPIConfigZ
model_name)aliasZpdx_model_nameZdevice_typeN	device_idTauto_configbackendbackend_confighpi_infoauto_paddle2onnx)__name__
__module____qualname__r	   strr   __annotations__r   r   intr   boolr   r   r   r   r   r   r   r    r   r   l/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/models/runners/hpi/config.pyr      s   
 r   c                   @   $   e Zd ZU edd dZeed< dS )OpenVINOConfigc                   C      t tddS NZPADDLE_PDX_CPU_NUM_THREADS
   r   osgetenvr   r   r   r   <lambda>)       zOpenVINOConfig.<lambda>default_factorycpu_num_threadsNr   r   r   r   r,   r   r   r   r   r   r   r!   '      
 r!   c                   @   r    )ONNXRuntimeConfigc                   C   r"   r#   r%   r   r   r   r   r(   /   r)   zONNXRuntimeConfig.<lambda>r*   r,   Nr-   r   r   r   r   r/   -   r.   r/   c                   @   sJ   e Zd ZU dZed ed< dZeed< dZe	e
eeee  f  ed< dS )TensorRTConfigfp32)r1   Zfp16	precisionTuse_dynamic_shapesNdynamic_shapes)r   r   r   r2   r   r   r3   r   r4   r   r   r   r   r   r   r   r   r   r0   3   s   
 $r0   c                   @   s   e Zd ZdS )OMConfigN)r   r   r   r   r   r   r   r5   9   s    r5   )r&   typingr   r   r   r   r   Zpydanticr   r   Ztyping_extensionsr	   r
   Zhpi.hpi_infor   r   r   r   r!   r/   r0   r5   r   r   r   r   <module>   s   
