
    {-j                         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
e          Z G d de          Z G d de          Z G d de          Z G d de          ZdS )    N)AnyDictListLiteralOptional)	BaseModelField)	Annotated	TypeAlias   )HPIInfo)paddleopenvinoonnxruntimetensorrtomInferenceBackendc                       e Zd ZU ee e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 )	HPIConfig
model_name)aliaspdx_model_name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        k/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddlex/inference/models/runners/hpi/config.pyr   r      s         c55|#<#<#<<====#Ix}###K*.GX&'.../3NHT#s(^,333"&Hhw&&&!d!!!!!r(   r   c                   4    e Zd ZU  ed           Zeed<   dS )OpenVINOConfigc                  F    t          t          j        dd                    S NPADDLE_PDX_CPU_NUM_THREADS
   r%   osgetenvr'   r(   r)   <lambda>zOpenVINOConfig.<lambda>)       BI.JB$O$O P P r(   default_factorycpu_num_threadsNr    r!   r"   r	   r7   r%   r$   r'   r(   r)   r+   r+   '   C          5PP  OS     r(   r+   c                   4    e Zd ZU  ed           Zeed<   dS )ONNXRuntimeConfigc                  F    t          t          j        dd                    S r-   r0   r'   r(   r)   r3   zONNXRuntimeConfig.<lambda>/   r4   r(   r5   r7   Nr8   r'   r(   r)   r;   r;   -   r9   r(   r;   c                   z    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)r?   fp16	precisionTuse_dynamic_shapesNdynamic_shapes)r    r!   r"   rA   r   r$   rB   r&   rC   r   r   r#   r   r%   r'   r(   r)   r>   r>   3   s_         )/Iw~&///####;?NHT#tDI"678?????r(   r>   c                       e Zd ZdS )OMConfigN)r    r!   r"   r'   r(   r)   rE   rE   9   s        Dr(   rE   )r1   typingr   r   r   r   r   pydanticr   r	   typing_extensionsr
   r   hpi.hpi_infor   r   r$   r   r+   r;   r>   rE   r'   r(   r)   <module>rJ      s   
				 5 5 5 5 5 5 5 5 5 5 5 5 5 5 % % % % % % % % 2 2 2 2 2 2 2 2 # # # # # #%9 )   
" " " " "	 " " "    Y       	   @ @ @ @ @Y @ @ @	 	 	 	 	y 	 	 	 	 	r(   