o
    â)i  ã                   @   s2   d Z ddlmZ ddlmZ deddfdd„ZdS )	z"
Worker-related helper functions.
é    )ÚSTR_NOT_IMPL_ENC_DEC_ERR_STRS)ÚGPUModelRunnerBaseÚ
enc_dec_mrÚreturnNc                 C   s¨   | j jr
ttd ƒ‚| jdurttd ƒ‚| jjrttd ƒ‚t| jj	ddƒdur/ttd ƒ‚| j
dur:ttd ƒ‚| jjdkrFttd	 ƒ‚| jjd
krRttd ƒ‚dS )zm
    Asserted that the provided encoder/decoder model runner instance reflects
    a supported scenario.
    Z!STR_NOT_IMPL_ENC_DEC_PREFIX_CACHENZSTR_NOT_IMPL_ENC_DEC_SWAZ$STR_NOT_IMPL_ENC_DEC_CHUNKED_PREFILLZattn_logit_softcappingZ"STR_NOT_IMPL_ENC_DEC_LOGIT_SOFTCAPZSTR_NOT_IMPL_ENC_DEC_LORAé   ZSTR_NOT_IMPL_ENC_DEC_PPr   ZSTR_NOT_IMPL_ENC_DEC_SPEC_DEC)Zcache_configZenable_prefix_cachingÚNotImplementedErrorr   Zsliding_windowZscheduler_configZchunked_prefill_enabledÚgetattrZmodel_configZ	hf_configZlora_configZparallel_configZpipeline_parallel_sizeZnum_lookahead_slots)r   © r	   ú]/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/vllm/worker/utils.pyÚ$assert_enc_dec_mr_supported_scenario   sB   
ÿ
ÿÿ
ÿÿÿ
ÿÿÿÿr   )Ú__doc__Z
vllm.utilsr   Zvllm.worker.model_runnerr   r   r	   r	   r	   r
   Ú<module>   s   ÿÿ