
    =pj	                         S r SSKJrJrJrJr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\5      r " S S\5      rg)zTransformers engine.    )AnyDictOptionalTupleType)	BaseModel
ConfigDict   )is_dep_available   )LocalModelFormat   )InferenceEnginec                      \ rS rSr% Sr\" SS9rSr\\	   \
S'   Sr\\	   \
S'   Sr\\   \
S'   Sr\\   \
S	'   Sr\\	   \
S
'   Sr\\\	\4      \
S'   Sr\\\	\4      \
S'   Sr\\\	\4      \
S'   Sr\\\	\4      \
S'   Srg)TransformersEngineConfig   z)Engine config for transformers inference.forbid)extraNdtypedevice_type	device_idtrust_remote_codeattn_implementationgeneration_configmodel_kwargsprocessor_kwargstokenizer_kwargs )__name__
__module____qualname____firstlineno____doc__r	   model_configr   r   str__annotations__r   r   intr   boolr   r   r   r   r   r   r   __static_attributes__r       i/var/www/html/pdf-tiff/venv/lib/python3.13/site-packages/paddlex/inference/models/engines/transformers.pyr   r      s    3H-LE8C=!%K#%#Ix}#(,x~,)-#-26xS#X/6-1L(4S>*115htCH~.515htCH~.5r*   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rg	)TransformersEngine+   z/Engine for Hugging Face Transformers inference.transformersreturnc                     g)Nr/   r   selfs    r+   nameTransformersEngine.name0   s    r*   c                     [         $ N)r   r2   s    r+   engine_config_model&TransformersEngine.engine_config_model4   s    ''r*   .c                     g)N)safetensorsr   r2   s    r+   get_supported_model_formats.TransformersEngine.get_supported_model_formats8   s      r*   N)
model_namedevicerawr>   r?   c                *    AU R                  X5        U$ r7   )_apply_device)r3   r@   r>   r?   s       r+   prepare_config_dict&TransformersEngine.prepare_config_dict=   s     3'
r*   c                 :    [        S5      (       d  [        S5      eg )Nr/   zXEngine 'transformers' is unavailable because dependency 'transformers' is not installed.)r   RuntimeErrorr2   s    r+   ensure_environment%TransformersEngine.ensure_environmentH   s$    //3  0r*   r   )r0   N)r   r    r!   r"   r#   entitiespropertyr%   r4   r   r   r8   r   r   r   r<   r   r   rC   rG   r)   r   r*   r+   r-   r-   +   s    9Hc   (T*B%C ( ( 	%(#-.	/  %) $	#s(^	 SM		
 	 
c3h	r*   r-   N)r#   typingr   r   r   r   r   pydanticr   r	   
utils.depsr   utils.model_pathsr   _baser   r   r-   r   r*   r+   <module>rP      s4      3 3 * + 0 "6y 6 " "r*   