o
    )i                      @   s`   d dl mZ d dlmZ d dlZd dlmZ ddlmZ dZ	ed Z
ed	d
G dd dZdS )    )	dataclass)LiteralN)Image   )get_vllm_public_assetsZvision_model_images)	stop_signcherry_blossomT)frozenc                   @   s>   e Zd ZU eed< edejfddZedej	fddZ
dS )
ImageAssetnamereturnc                 C   s   t | j dtd}t|S )Nz.jpgfilenameZ	s3_prefix)r   r   VLM_IMAGES_DIRr   openselfZ
image_path r   ]/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/vllm/assets/image.py	pil_image   s   
zImageAsset.pil_imagec                 C   s$   t | j dtd}tj|dddS )zR
        Image embeddings, only used for testing purposes with llava 1.5.
        z.ptr   cpuT)Zmap_locationZweights_only)r   r   r   torchloadr   r   r   r   image_embeds   s   zImageAsset.image_embedsN)__name__
__module____qualname__ImageAssetName__annotations__propertyr   r   r   ZTensorr   r   r   r   r   r
      s   
 r
   )dataclassesr   typingr   r   ZPILr   baser   r   r   r
   r   r   r   r   <module>   s   