{-j UdZddlmZddlmZddlmZddlmZm Z m Z m Z m Z m Z ddlmZddlmZed Zd ed <e e ed fe eZd ed<Gdde dZefddZefddZdS)zPModel file paths and format resolution for inference model directory convention.) annotations)PathLike)Path)LiteralTuple TypedDictUnioncastget_args) TypeAlias)MODEL_FILE_PREFIX)paddleonnxom paddle_dyn safetensorsr LocalModelFormat.zTuple[LocalModelFormat, ...]LOCAL_MODEL_FORMATScFeZdZUdZded<ded<ded<ded<ded<d S) ModelPathsz7Resolved local model files keyed by `LocalModelFormat`.zTuple[Path, Path]rrrrrrN)__name__ __module__ __qualname____doc____annotations__j/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddlex/inference/models/utils/model_paths.pyrr(sNAAJJJ HHHrrF)total model_dirUnion[str, PathLike]model_file_prefixstrreturncVt|}i}d}||dz r ||dz }n"||dz r||dz }|r'||dz r |||dz f|d<||dz r ||dz |d<||dz r ||dz |d<|d z r|d z |d <|d z r|d z |d <|d z r|d z |d <|S)Nz.jsonz.pdmodelz .pdiparamsrz.onnxrz.omrzmodel_state.pdparamsrzinference.pdparamszmodel.safetensorsr)rexists)r!r# model_paths pd_model_paths rget_model_pathsr*2sYI KM(////7799C!'8$?$?$?? *444 4 < < > >C!'8$B$B$BB  )):&F&F&FFNNPP  ,888 8! H (////7799F'->*E*E*EE F(----5577B%+<(A(A(AA D**2244G$-0F$F L!((0022E$-0D$D L!''//11E%.1D%D M" rUnion[str, PathLike, Path] str | Nonecdtt||}d|vrdSd|vrdSd|vrdSdS)zResolve paddle_static vs paddle_dynamic from actual model files. Returns: "paddle_static", "paddle_dynamic", or None if no paddle model files found. r paddle_staticrpaddle_dynamicrN)r*r)r!r#pathss r&resolve_paddle_engine_from_model_filesr1OsQ DOO-> ? ?E5u 4rN)r!r"r#r$r%r)r!r+r#r$r%r,)r __future__rosrpathlibrtypingrrrr r r typing_extensionsr constantsrrrrrr*r1rrrr8syWVV""""""CCCCCCCCCCCCCCCC''''''++++++%59D C  ((+;"<"<55 %/>/r