o !iD@slddlmZddlmZmZddlmZmZmZm Z m Z m Z m Z m Z ddlmZddZdd Zd d Zd S) )PaddlePredictorOption)get_default_device parse_device)DEFAULT_CPU_THREADSDEFAULT_DEVICEDEFAULT_ENABLE_MKLDNNDEFAULT_MKLDNN_CACHE_CAPACITYDEFAULT_PRECISIONDEFAULT_USE_TENSORRTSUPPORTED_PRECISION_LISTDEFAULT_USE_CINN)str2boolc Cst|ttttttd}||}|D]}td|i||}|dt vr7td|ddt d| d|d<| d|d <|S) N)device enable_hpi use_tensorrt precision enable_mkldnnmkldnn_cache_capacity cpu_threads enable_cinnzUnknown argument: rzInvalid precision: z. Supported values are: .r use_pptrtpptrt_precision) rr r rr rr keys ValueErrorr pop)kwargsdefault_enable_hpiZ default_valsZ unknown_namesnamer b/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/paddleocr/_common_args.pyparse_common_argss(   r"cCs|d}|dur t}t|\}}i}||d<|d|d<t}|dkrF|drB|ddkr2d|_n0|dd ks>J|dd |_n d |_n|d kr_|d }|rV|d|_nd |_|d|_nd |_|d|_||d<|S)NrrZuse_hpipZgpurrZfp32Ztrt_fp32Zfp16Ztrt_fp16Zpaddlecpurrrr pp_option)rrrZrun_moderrr)Z model_nameZ common_argsrZ device_type_Z init_kwargsr$rr r r!prepare_common_init_args<s6      r&cCs|rd}nd}|jdtt|d|jdt|dd|jdttdd|jd tttd d |jd ttd d|jdtt dd|jdtt dd|jdtt dddS)NaDevice(s) to use for inference, e.g., `cpu`, `gpu`, `npu`, `gpu:0`, `gpu:0,1`. If multiple devices are specified, inference will be performed in parallel. Note that parallel inference is not always supported. By default, GPU 0 will be used if available; otherwise, the CPU will be used.zDevice to use for inference, e.g., `cpu`, `gpu`, `npu`, `gpu:0`. By default, GPU 0 will be used if available; otherwise, the CPU will be used.z--device)typedefaulthelpz --enable_hpiz&Enable the high performance inference.z--use_tensorrtzWhether to use the Paddle Inference TensorRT subgraph engine. If the model does not support TensorRT acceleration, even if this flag is set, acceleration will not be used.z --precisionzPPrecision for TensorRT when using the Paddle Inference TensorRT subgraph engine.)r'r(choicesr)z--enable_mkldnnzEnable MKL-DNN acceleration for inference. If MKL-DNN is unavailable or the model does not support it, acceleration will not be used even if this flag is set.z--mkldnn_cache_capacityzMKL-DNN cache capacity.z --cpu_threadsz/Number of threads to use for inference on CPUs.z --enable_cinnz!Whether to use the CINN compiler.) add_argumentstrrrr r r rintr rr )parserrZallow_multiple_deviceshelp_r r r!add_common_cli_optsash r0N)Zpaddlex.inferencerZpaddlex.utils.devicerr _constantsrrrr r r r r Z _utils.clirr"r&r0r r r r!s (  %