jjddlmZmZddlmZmZmZmZmZm Z m Z m Z ddl m Z gdZdZdZdZd Zd S) )get_default_device parse_device)DEFAULT_CPU_THREADSDEFAULT_DEVICEDEFAULT_ENABLE_MKLDNNDEFAULT_MKLDNN_CACHE_CAPACITYDEFAULT_PRECISIONDEFAULT_USE_TENSORRTSUPPORTED_PRECISION_LISTDEFAULT_USE_CINN)str2bool)paddle paddle_staticpaddle_dynamic transformersc  tdd|ttttt t d }||z }|D]}td|i||}|d0|dtvr!td|ddtd|dtvr!td|ddtd| d |d <| d|d <|S) N) deviceengine engine_config enable_hpi use_tensorrt precision enable_mkldnnmkldnn_cache_capacity cpu_threads enable_cinnzUnknown argument: rzInvalid engine: z. Supported values are: .rzInvalid precision: r use_pptrtpptrt_precision) rr r rr rr keys ValueErrorSUPPORTED_INFERENCE_ENGINE_LISTr pop)kwargsdefault_enable_hpi default_vals unknown_namesnames f/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/paddleocr/_common_args.pyparse_common_argsr+%sB (,&.!>*'  LKKMML$5$5$7$77M664d44555 ' ' 'F x$ 8 $C C C kvh/ k kIh k k k   k"::: j&"5 j jOg j j j   !**^44F; & ; 7 7F  Mci}|dkrB|dr4|ddkrd|d<nX|ddksJ|dd|d<n6d |d<n0|d kr%|d r |d |d <nd |d<|d |d <nd |d<|d|d<|S)Ngpurr fp32trt_fp32run_modefp16trt_fp16rcpurrrr) common_args device_typecfgs r*"_build_paddle_static_engine_configr9Ls Ce { # ',-77",J"#45???%B???#-J&C OO     ' '+67N+OC' ( (&C O(7M"J$]3C  Jr,c|d}|t}t|\}}i}||d<|d|d<|d|d<|d}|d}t||}|||d<n|dkr||d<n|dvrd|i|d<nd|d<|S)Nrrruse_hpiprr)Nr)rrr9) model_namer6rr7_ init_kwargsuser_engine_configrbuilts r*prepare_common_init_argsrAes  "F ~#%%!&))NKK"K'1K),7K $_5  "F .{K H HE%'9 O$$ ? " "', O$$ # # #(7'? O$$'+ O$ r,c|rd}nd}|dtt||dttd|dt|d |d tt d |d tt td |dttd|dttd|dttd|dttddS)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--engineznInference engine to use. For CLI, engine-specific configuration should be set in the PaddleX YAML config file.)rCchoicesrEz --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.)rCrDrFrEz--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_argumentstrrr#rr r r rintr rr )parserr&allow_multiple_deviceshelp_s r*add_common_cli_optsrMsaqa       / }    " 5    ${    !( _   %n   ! - &    # >     0 r,N)paddlex.utils.devicerr _constantsrrrr r r r r _utils.clirr#r+r9rArMr5r,r*rQsBAAAAAAA                    ! ###$$$N26;;;;;r,