+hj" SSKrSSKrSSKJr SSKJrJr SSKrSSK J r SSK J r SSK JrJrJrJr \"5(aSSKr\R("\5r\R.\R0\R2\R4\R6\R8\R:\R<\R>\R@\RB\RDS. r#"S S 5r$g) N)Path)OptionalUnion)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~\rSrSrSSjrSr\SS\\\ 44Sjj5r SS\\\ 4S\ \4S jjr S\\\ R44S jr\\SS \\\ 4S \ \\\S4S \ \\S4S\S\ \S\ \S\ \S\ S4Sjj55r\\SS \\\ 4S\S \ \S\ \4Sjj55rSrg)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/james-whalen/.local/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#ApathcUc[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)r(providerr,r-s r load_modelOnnxRuntimeModel.load_model?sY   KKW X-H  #! ,d33 01 ## J\  rsave_directory file_namec 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 SameFileErrorr exists)rr5r6rmodel_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)osr(isfilererrormakedirsr@)rr5rs rsave_pretrained OnnxRuntimeModel.save_pretrainedvsT 77>>. ) ) LL?>*::]^ _  NT2 n77rmodel_idtokenrevisionforce_download cache_dirr2r,zort.SessionOptionsc 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 (`Union[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-)r2r,r-r)repo_idfilenamerKrLrNrMrr) r rDr(isdirrr3ras_posixpoprparentname) clsrJrKrLrMrNr6r2r,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)rJrLrNrMrKrR)lenstrsplitrZ)rXrJrMrKrN model_kwargsrLs rfrom_pretrained OnnxRuntimeModel.from_pretrainedsg s8}""3' (A -!)!4 H## )     r)rrrr)NNN)NNFNNNN)TNN)__name__ __module__ __qualname____firstlineno__rr& staticmethodrr_rr3rr@rDPathLikerH classmethodrboolrZrb__static_attributes__rRrrrr4sT , sDy)  .uS$Y/?HUXM>8c2;;./8,37/3$#'#'"&7;A*T "A*dCo./A*5d+, A*  A* C= A*C=A*3-A*34A*A*F $##'  T "  }  C=   rr)%rDr9pathlibrtypingrrnumpyr huggingface_hubrhuggingface_hub.utilsrutilsr r r r onnxruntimer0 get_loggerrdrbool_int8uint8int16uint16int32uint32int64uint64float16float32float64ORT_TO_NP_TYPErrRrrrs$ "+6]]   H %HHGGXXXXiiXXiiXXiizzZZjj  r r r