wi VddlZddlZddlmZddlmZmZddlZddl m Z ddl m Z ddl mZmZmZmZerddlZejeZejejejejejejejejejej ej!ej"d Z#Gd d Z$dS) 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)ceZdZddZdZeddeeeffdZ ddeeefde efdZ deee j ffd Zee dd eeefd e eeedfd e eedfdede ede ede ede dfdZee dd eeefded e ede efdZdS)OnnxRuntimeModelNc td||_|dd|_|dt |_dS)NzL`diffusers.OnnxRuntimeModel` is experimental and might change in the future.model_save_dirlatest_model_name)loggerinfomodelgetrr r)selfrkwargss n/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/diffusers/pipelines/onnx_utils.py__init__zOnnxRuntimeModel.__init__5sQ bccc $jj)94@@!',?AR!S!Sc td|D}|jd|S)Nc>i|]\}}|tj|S)nparray).0kvs r z-OnnxRuntimeModel.__call__..<s&<<=C>c tj|r td|ddStj|d|j|fi|dS)a$ 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<)rr1rs rsave_pretrainedz OnnxRuntimeModel.save_pretrainedosu 7>>. ) )  LL^>^^^ _ _ _ F NT2222 n7777777rFmodel_idtokenrevisionforce_download cache_dirr/r,zort.SessionOptionsc  ||nt} tj|rPtt || ||} t || d<n`t|| ||||} t | j | d<t | j | d<t| ||} |dd| i| S)aA 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 N)r/r,r)repo_idfilenamerErFrHrGrrr) r r?r(isdirrr0ras_posixrparentname) clsrDrErFrGrHr2r/r,rr9rmodel_cache_paths r_from_pretrainedz!OnnxRuntimeModel._from_pretraineds L(1'<))BS 7== " " p$//X//88::X\h0E(,H~~F# $ $ / (!#-     (,,<'='='DF# $*./?*@*@*EF& '$//0@8bn/ooEs)))&)))rTc d}tt|ddkr|d\}}|jd|||||d|S)N@r)rDrFrHrGrEr)lenstrsplitrR)rPrDrGrErH model_kwargsrFs rfrom_pretrainedz OnnxRuntimeModel.from_pretraineds s8}}""3'' ( (A - -!)!4!4 Hh#s# )       r)N)NN)NNFNNNN)TNN)__name__ __module__ __qualname__rr' staticmethodrrVrr0rr<r?PathLikerC classmethodrboolrRrYrrrrr4sTTTT ,,,[[sDy)[[[\[ uS$Y/?HUXM>8c2;./8888,37/3$#'#'"&7;9*9*T "9*dCo./9*5d+, 9*  9* C= 9*C=9*3-9*349*9*9*[9*v $##'   T "  }  C=    [   rr)%r?r5pathlibrtypingrrnumpyrhuggingface_hubrhuggingface_hub.utilsrutilsr r r r onnxruntimer- get_loggerrZrbool_int8uint8int16uint16int32uint32int64uint64float16float32float64ORT_TO_NP_TYPErrrrrvsP$ """"""""++++++666666]]]]]]]]]]]]  H % %HGXXiXiXizZj   c c c c c c c c c c r