3j " SSKJr SSKrSSKrSSKJr SSKrSSKJ r SSK J r SSK J r JrJrJr \"5(aSSKr\R&"\5r\R,\R.\R0\R2\R4\R6\R8\R:\R<\R>\R@\RBS. r""S S 5r#g) ) annotationsN)Path)hf_hub_download)validate_hf_hub_args)ONNX_EXTERNAL_WEIGHTS_NAMEONNX_WEIGHTS_NAMEis_onnx_availablelogging) z tensor(bool)z tensor(int8)z tensor(uint8)z tensor(int16)ztensor(uint16)z tensor(int32)ztensor(uint32)z tensor(int64)ztensor(uint64)ztensor(float16)z tensor(float)ztensor(double)c\rSrSrS SjrSr\S S Sjj5rS SSjjrSSjr \ \ SSSjj55r \ \ SSS jj55r S rg)OnnxRuntimeModel4Nc [RS5 XlURSS5UlURS[ 5Ulg)NzL`diffusers.OnnxRuntimeModel` is experimental and might change in the future.model_save_dirlatest_model_name)loggerinfomodelgetrr r)selfrkwargss X/home/wildlama/miniconda3/lib/python3.13/site-packages/diffusers/pipelines/onnx_utils.py__init__OnnxRuntimeModel.__init__5s= bc $jj)94@!',?AR!Sc UR5VVs0sHup#U[R"U5_M nnnURR SU5$s snnfN)itemsnparrayrrun)rrkvinputss r__call__OnnxRuntimeModel.__call__;sD-3\\^<^TQ!RXXa[.^<zz~~dF++=s#AcUc[RS5 SnUc/nO[U[5(dU/n[R "X/X#S9$)a= Loads an ONNX Inference session with an ExecutionProvider. Default provider is `CPUExecutionProvider` Arguments: path (`str` or `Path`): Directory from which to load provider(`str`, *optional*): Onnxruntime execution provider to use for loading the model, defaults to `CPUExecutionProvider` z=No onnxruntime provider specified, using CPUExecutionProviderCPUExecutionProvider) providers sess_optionsprovider_options)rr isinstancelistortInferenceSession)pathproviderr*r+s r load_modelOnnxRuntimeModel.load_model?sY   KKW X-H  #! ,d33 01 ## J\  rc UbUO[nURRUR5n[ U5RU5n[ R "XV5 URR[5nUR5(a6[ U5R[5n[ R "XV5 gg![ Ra Nf=f![ Ra gf=f)a Save a model and its configuration file to a directory, so that it can be re-loaded using the [`~optimum.onnxruntime.modeling_ort.ORTModel.from_pretrained`] class method. It will always save the latest_model_name. Arguments: save_directory (`str` or `Path`): Directory where to save the model file. file_name(`str`, *optional*): Overwrites the default model file name from `"model.onnx"` to `file_name`. This allows you to save the model with a different name. N) r rjoinpathrrshutilcopyfile SameFileErrorrexists)rsave_directory file_namermodel_file_namesrc_pathdst_paths r_save_pretrained!OnnxRuntimeModel._save_pretrainedWs(1'<)BS&&//0F0FG'00A  OOH / &&//0JK ??  N+445OPH 3  ##   ''  s$ C 5C' C$#C$'C>=C>c [RRU5(a[R SUS35 g[R "USS9 UR "U40UD6 g)z Save a model to a directory, so that it can be re-loaded using the [`~OnnxModel.from_pretrained`] class method.: Arguments: save_directory (`str` or `os.PathLike`): Directory to which to save. Will be created if it doesn't exist. zProvided path (z#) should be a directory, not a fileNT)exist_ok)osr0isfilererrormakedirsr?)rr:rs rsave_pretrained OnnxRuntimeModel.save_pretrainedvsT 77>>. ) ) LL?>*::]^ _  NT2 n77rc UbUO[n [RRU5(aK[R [ X5R5UUU RS5S9n [ U5U S'Oc[UU UUUUS9n [ U 5RU S'[ U 5RU S'[R U UUU RS5S9n U"SSU 0U D6$)a Load a model from a directory or the HF Hub. Arguments: model_id (`str` or `Path`): Directory from which to load token (`str` or `bool`): Is needed to load models from a private or gated repository revision (`str`): Revision is the specific model version to use. It can be a branch name, a tag name, or a commit id cache_dir (`str | Path`, *optional*): Path to a directory in which a downloaded pretrained model configuration should be cached if the standard cache should not be used. force_download (`bool`, *optional*, defaults to `False`): Whether or not to force the (re-)download of the model weights and configuration files, overriding the cached versions if they exist. file_name(`str`): Overwrites the default model file name from `"model.onnx"` to `file_name`. This allows you to load different model files from the same repository or directory. provider(`str`): The ONNX runtime provider, e.g. `CPUExecutionProvider` or `CUDAExecutionProvider`. kwargs (`Dict`, *optional*): kwargs will be passed to the model during initialization r+)r1r*r+r)repo_idfilenametokenrevision cache_dirforce_downloadrr) r rCr0isdirr r2ras_posixpoprparentname) clsmodel_idrLrMrOrNr;r1r*rr<rmodel_cache_paths r_from_pretrained!OnnxRuntimeModel._from_pretrainedsL(1'<)BS 77== " "$//X/88:!)!',>!? 0E (,H~F# $ / (!#-   (,,<'='D'DF# $*./?*@*E*EF& '$// !)!',>!? 0E ))&))rc Sn[[U5RS55S:XaURS5upUR"SUUUUUS.UD6$)N@r)rWrMrNrOrLrP)lenstrsplitrY)rVrWrOrLrN model_kwargsrMs rfrom_pretrained OnnxRuntimeModel.from_pretrainedsg s8}""3' (A -!)!4 H## )     r)rrrr)NNN)r0 str | Path)r:rcr; str | None)r:zstr | os.PathLike)NNFNNNN)rWrcrLzbool | str | None | NonerMzstr | None | NonerOboolrNrdr;rdr1rdr*z'ort.SessionOptions' | None)TNN)rWrcrOrerLrdrNrd)__name__ __module__ __qualname____firstlineno__rr% staticmethodr2r?rG classmethodrrYra__static_attributes__rPrrr r 4s T ,  .>8)8,+/&*$ $ $#48A*A*(A*$ A*  A*  A*A*A*2A*A*F $ $         rr )$ __future__rrCr6pathlibrnumpyrhuggingface_hubrhuggingface_hub.utilsrutilsrr r r onnxruntimer. get_loggerrfrbool_int8uint8int16uint16int32uint32int64uint64float16float32float64ORT_TO_NP_TYPEr rPrrrs"# +6]]   H %HHGGXXXXiiXXiiXXiizzZZjj  r r r