
    =pjg                        S r SSKJr  SSKJrJrJrJrJr  SSK	J
r
  SSKJr  SSKJr  SS	KJr  SS
KJrJr  SSKJr  SSKJr  SSKJr  SSKJr  SSKJrJr  S\\\\4      S\\\4   S\\\4   4S jr  " S S\5      r! " S S\5      r"g)zPaddle engines.    )Path)AnyDictOptionalTupleType   )MODEL_FILE_PREFIX)logging)is_dep_available   )get_hpi_info)PaddleDynamicRunnerPaddleStaticRunner)InferenceRunnerPaddleDynamicRunnerConfigPaddleStaticRunnerConfig)LocalModelFormat   )RunnerBuilderRunnerEnginemodel_configengine_configreturnc                    [        U 5      nUc  U$ S nUR                  (       a  UR                  R                  nUc  U$ UR                  S5      c@  UR                  b3  [
        R                  " SUR                  5        0 UESUR                  0EnUR                  S5      c@  UR                  b3  [
        R                  " SUR                  5        0 UESUR                  0EnU$ )Ntrt_dynamic_shapesz!TensorRT dynamic shapes set to %strt_dynamic_shape_input_dataz+TensorRT dynamic shape input data set to %s)r   backend_configspaddle_infergetr   r   debugr   )r   r   hpi_infopaddle_infos       c/var/www/html/pdf-tiff/venv/lib/python3.13/site-packages/paddlex/inference/models/engines/paddle.py_inject_trt_infor'       s     L)HK..;;./7**6/1O1O	


 +"@"@

 	89A44@944	


*K,T,T
     c                       \ rS rSrSrSr\S\4S j5       r\S\	\
   4S j5       rSSS.S	\\\4   S
\\   S\\   S\\\4   4S jjrSS jrS\4S jrS\SS4S jrSrg)PaddleStaticEngineF   z)Engine for Paddle static-graph inference.paddle_staticr   c                     g)Nr,    selfs    r&   namePaddleStaticEngine.nameK   s    r(   c                     [         $ Nr   r/   s    r&   engine_config_model&PaddleStaticEngine.engine_config_modelO   s    ''r(   N
model_namedevicerawr8   r9   c                    A[        [        R                  5      nUR                  5        VVs0 s H  u  pVXT;   d  M  XV_M     nnnU R	                  X5        U$ s  snnf r4   )setr   model_fieldsitems_apply_device)r0   r:   r8   r9   valid_fieldskeyvalues          r&   prepare_config_dict&PaddleStaticEngine.prepare_config_dictS   sV     3@@A,/IIKOKjc3;NzszKO3'
 Ps
   AAc                 :    [        S5      (       d  [        S5      eg )NpaddlepaddlezYEngine 'paddle_static' is unavailable because dependency 'paddlepaddle' is not installed.r   RuntimeErrorr/   s    r&   ensure_environment%PaddleStaticEngine.ensure_environment`   $    //3  0r(   c                     S S.S[         S[        [           S[        [        [         [        4      S[        [         [        4   S[        [
           S[        4S jjnU$ )	N)default_builderr8   	model_dirr   r   rM   r   c                 j    AUc  [        S5      e[        U[        U5      5      n[        U U[        US9$ )Nz3`model_dir` is required for engine='paddle_static'.)r8   rN   model_file_prefixconfig)
ValueErrorr'   dictr   r
   )r8   rN   r   r   rM   runner_configs         r&   runner_builderEPaddleStaticEngine.get_default_runner_builder.<locals>.runner_builderh   sC        !VWW,\4;NOM%%#"3$	 r(   )strr   r   r   r   r   r   )r0   rU   s     r&   get_default_runner_builder-PaddleStaticEngine.get_default_runner_builderg   sl     8<		  ~	 #4S>2		
  S>	 &m4	 	& r(   runnerc                 r    [        U[        5      (       d"  [        S[        U5      R                   S35      eg )Nz@Engine 'paddle_static' must build a PaddleStaticRunner, but got .)
isinstancer   	TypeErrortype__name__r0   rZ   s     r&   validate_runner"PaddleStaticEngine.validate_runner}   s>    &"455<0014  6r(   r.   r   N)r`   
__module____qualname____firstlineno____doc__entitiespropertyrW   r1   r   r   r5   r   r   r   rC   rI   r   rX   r   rb   __static_attributes__r.   r(   r&   r*   r*   F   s    3Hc   (T*B%C ( ( %) $#s(^ SM	
  
c3hM ,o $ r(   r*   c                       \ rS rSrSrSr\S\4S j5       r\S\	\
   4S j5       rS\\\S4      4S jrS	S	S
.S\\\4   S\\   S\\   S\\\4   4S jjrSS jrS\\   4S jrS\SS	4S jrSrg	)PaddleDynamicEngine   z*Engine for Paddle dynamic-graph inference.paddle_dynamicr   c                     g)Nro   r.   r/   s    r&   r1   PaddleDynamicEngine.name   s    r(   c                     [         $ r4   r   r/   s    r&   r5   'PaddleDynamicEngine.engine_config_model   s    ((r(   .c                     g)N)safetensors
paddle_dynr.   r/   s    r&   get_supported_model_formats/PaddleDynamicEngine.get_supported_model_formats   s     -r(   Nr7   r:   r8   r9   c                *    AU R                  X5        U$ r4   )r?   )r0   r:   r8   r9   s       r&   rC   'PaddleDynamicEngine.prepare_config_dict   s     3'
r(   c                 :    [        S5      (       d  [        S5      eg )NrF   zZEngine 'paddle_dynamic' is unavailable because dependency 'paddlepaddle' is not installed.rG   r/   s    r&   rI   &PaddleDynamicEngine.ensure_environment   rK   r(   c                     g r4   r.   r/   s    r&   rX   .PaddleDynamicEngine.get_default_runner_builder   s    r(   rZ   c                 r    [        U[        5      (       d"  [        S[        U5      R                   S35      eg )NzBEngine 'paddle_dynamic' must build a PaddleDynamicRunner, but got r\   )r]   r   r^   r_   r`   ra   s     r&   rb   #PaddleDynamicEngine.validate_runner   s>    &"566<0014  7r(   r.   rd   )r`   re   rf   rg   rh   ri   rj   rW   r1   r   r   r5   r   r   r   rw   r   r   rC   rI   r   rX   r   rb   rk   r.   r(   r&   rm   rm      s    4H c     )T*C%D ) )-	%(#-.	/- %) $	#s(^	 SM		
 	 
c3h	H],C o $ r(   rm   N)#rh   pathlibr   typingr   r   r   r   r   	constantsr
   utilsr   
utils.depsr   hpir   runnersr   r   runners.inference_runnerr   runners.paddle_dynamic_runnerr   runners.paddle_staticr   utils.model_pathsr   _baser   r   rW   r'   r*   rm   r.   r(   r&   <module>r      s       3 3 +  +  = 6 E < 0 .#4S>*#S># 
#s(^#L< <~,, ,r(   