/;jio;|ddlmZgdZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z mZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2dd l3m4Z4m5Z5dd l6m7Z7m8Z8m9Z9ddl:Zdd lm;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFejGrdd lHmIZIejJjJZKed eLzZMdZNdCdZOdDdZPdEdZQ dFdGdZR dHdId$ZS dFdJd&ZTeNfdKd*ZUeNfdLd+ZV dFd,ddd-d,d.dMd7ZW dFdNd8ZXeSZYeUZZeWZ[dOd;Z\dPd=Z]dQd?Z^dRdBZ_e\e!_`e]e_`e^e_`dS)S) annotations)>ONNX_ML IR_VERSIONIR_VERSION_2017_10_10IR_VERSION_2017_10_30IR_VERSION_2017_11_3IR_VERSION_2019_1_22IR_VERSION_2019_3_18IR_VERSION_2019_9_19IR_VERSION_2020_5_8IR_VERSION_2021_7_30IR_VERSION_2023_5_5IR_VERSION_2024_3_25 EXPERIMENTALSTABLEcheckercomposedefs gen_protohelperhub numpy_helperparserprintershape_inferenceutilsversion_converterAttributeProtoDeviceConfigurationProto FunctionProto GraphProtoIntIntListEntryProtoMapProto ModelProtoNodeDeviceConfigurationProto NodeProto OperatorProtoOperatorSetIdProtoOperatorSetProtoOperatorStatus OptionalProto SequenceProtoSimpleShardedDimProtoShardedDimProtoShardingSpecProtoSparseTensorProtoStringStringEntryProtoTensorAnnotation TensorProtoTensorShapeProtoTrainingInfoProto TypeProtoValueInfoProtoVersionconvert_model_to_external_dataload_external_data_for_modelload_model_from_string load_modelload_tensor_from_string load_tensor save_model save_tensorwrite_external_data_tensorsN)IOLiteral) serialization)r)r:rAr9)#rrrr r!r"rrrrr r r r r rrr$r%r&r(r*rr-r.r/r0r1r2r3r4r5r6r7r8)r'r))r#r+r,) rrrrrrrrrrrr)Sequence)protobuf textprotoonnxtxtjsonrFfIO[bytes] | str | os.PathLikereturnbytescft|drOttjd|jr(tjd|}nQtjd|}t |d5}|}dddn #1swxYwY|S)Nread IO[bytes]str | os.PathLikerb)hasattrcallabletypingcastrOopen)rJcontentreadables G/root/voice-cloning/.venv/lib/python3.11/site-packages/onnx/__init__.py _load_bytesr[sq&&hv{;'B'B'GHH&+k1--2244 K+Q / / !T]] &hmmooG & & & & & & & & & & & & & & & NsB&&B*-B*rXNonecnt|drQttjd|jr*tjd||dStjd|}t |d5}||ddddS#1swxYwYdS)NwriterPrQwb)rSrTrUrVr^rW)rXrJwritables rZ _save_bytesrasq'$x K(C(C(IJJ$ K##))'22222 K+Q / / !T]] $h NN7 # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $sB**B.1B.$IO[bytes] | str | os.PathLike | None str | Nonect|ttjfrtj|St |dr(|Jtj|jSdS)Nname) isinstancestrosPathLikepathabspathrSre)rJs rZ_get_file_pathrlse!c2;'(("wq!!!q&'}}}wqv&&& 4fmt_SupportedFormat | None$str | os.PathLike | IO[bytes] | Noneserialization.ProtoSerializerc@|tj|St|x}Atj|\}}tj|}|pt}|Jtj|S)zQGet the serializer for the given path and format from the serialization registry.) rDregistrygetrlrhrjsplitextget_format_from_file_extension_DEFAULT_FORMAT)rnrJ file_path_exts rZ_get_serializerr{s  %))#...#A&&& 3!!),,3$CCCHH  C ???  ! % %c * **rmTformatload_external_databoolr$ct||t|t}|r@t |}|r/t j|}t|||S)aLoads a serialized ModelProto into memory. Args: f: can be a file-like object (has "read" function) or a string/PathLike containing a file name format: The serialization format. When it is not specified, it is inferred from the file extension when ``f`` is a path. If not specified _and_ ``f`` is not a path, 'protobuf' is used. The encoding is assumed to be "utf-8" when the format is a text format. load_external_data: Whether to load the external data. Set to True if the data is under the same directory of the model. If not, users need to call :func:`load_external_data_for_model` with directory to load external data from. Returns: Loaded in-memory ModelProto. ) r{deserialize_protor[r$rlrhrjdirnamer:)rJr|r}modelmodel_filepathbase_dirs rZr<r<sr* FA & & 8 8Q V VE:'**  :w~66H ( 9 9 9 Lrmr3c|t||t|tS)aLoads a serialized TensorProto into memory. Args: f: can be a file-like object (has "read" function) or a string/PathLike containing a file name format: The serialization format. When it is not specified, it is inferred from the file extension when ``f`` is a path. If not specified _and_ ``f`` is not a path, 'protobuf' is used. The encoding is assumed to be "utf-8" when the format is a text format. Returns: Loaded in-memory TensorProto. )r{rr[r3)rJr|s rZr>r>s. 61 % % 7 7 A V VVrms bytes | str_SupportedFormatc`t||tS)aLoads a binary string (bytes) that contains serialized ModelProto. Args: s: a string, which contains serialized ModelProto format: The serialization format. When it is not specified, it is inferred from the file extension when ``f`` is a path. If not specified _and_ ``f`` is not a path, 'protobuf' is used. The encoding is assumed to be "utf-8" when the format is a text format. Returns: Loaded in-memory ModelProto. )r{rr$rr|s rZr;r;s& 6 " " 4 4Q E EErmc`t||tS)aLoads a binary string (bytes) that contains serialized TensorProto. Args: s: a string, which contains serialized TensorProto format: The serialization format. When it is not specified, it is inferred from the file extension when ``f`` is a path. If not specified _and_ ``f`` is not a path, 'protobuf' is used. The encoding is assumed to be "utf-8" when the format is a text format. Returns: Loaded in-memory TensorProto. )r{rr3rs rZr=r=s& 6 " " 4 4Q F FFrmFi)save_as_external_dataall_tensors_to_one_filelocationsize_thresholdconvert_attributeprotoModelProto | bytesrrrrintrct|tr4tt|t }|rt |||||t|}|/tj |} t|| }t|| |} t| |dS)a}Saves the ModelProto to the specified path and optionally, serialize tensors with raw data as external data before saving. Args: proto: should be a in-memory ModelProto f: can be a file-like object (has "write" function) or a string containing a file name or a pathlike object format: The serialization format. When it is not specified, it is inferred from the file extension when ``f`` is a path. If not specified _and_ ``f`` is not a path, 'protobuf' is used. The encoding is assumed to be "utf-8" when the format is a text format. save_as_external_data: If true, save tensors to external file(s). all_tensors_to_one_file: Effective only if save_as_external_data is True. If true, save all tensors to one external file specified by location. If false, save each tensor to a file named with the tensor name. location: Effective only if save_as_external_data is true. Specify the external file that all tensors to save to. Path is relative to the model path. If not specified, will use the model name. size_threshold: Effective only if save_as_external_data is True. Threshold for size of data. Only when tensor's data is >= the size_threshold it will be converted to external data. To convert every tensor with raw data to external data set size_threshold=0. convert_attribute: Effective only if save_as_external_data is True. If true, convert all tensors to external data If false, convert only non-attribute tensors to external data N)rfrMr{rwrr$r9rlrhrjrrAserialize_protora) rrJr|rrrrrrbasepath serializeds rZr?r?'sH%X00BB5*,,WW & *HnFW   $A&&N!7??>22+E8<< 88HHOOJ Armclt|||}t||dS)aSaves the TensorProto to the specified path. Args: proto: should be a in-memory TensorProto f: can be a file-like object (has "write" function) or a string containing a file name or a pathlike object. format: The serialization format. When it is not specified, it is inferred from the file extension when ``f`` is a path. If not specified _and_ ``f`` is not a path, 'protobuf' is used. The encoding is assumed to be "utf-8" when the format is a text format. N)r{rra)rrJr|rs rZr@r@\s7 !++;;EBBJ Armselfrgc \|jr d|jd}nd}|jr d|jd}nd}|jr d|jd}nd}|jr d|j}nd}|jrdt |jd}nd}|jrd t|j}nd}d |j||||||d S) N , domain=''z, producer_name='z, producer_version='z, graph=z , functions=, opset_import=zModelProto(ir_version=)) domain producer_nameproducer_versiongraph functionslen opset_import_operator_set_protos_repr ir_version)rrrrrrrs rZ_model_proto_reprrvs {,dk,,, AD,>AAA   J$2GJJJ z)4:)) ~DC$7$7DDD   W)B4CT)U)UWW  DO \ 6 = Zj lq s|   rmr!c |jrdt|jd}nd}|jrdt|jd}nd}|jrdt|jd}nd}|jrdt|jd }nd}|jrd t|jd }nd}d |jd |||||d S)Nz, initializer=r, node=< nodes>z, value_info= , input=< inputs> , output=< outputs>z GraphProto('rr) initializerrnode value_infoinputoutputre)rrrrrrs rZ_graph_proto_reprrs MD,<(=(=MMM   y1#di..111 Hc$/&:&:HHH   z5C OO555 {9c$+..999 U$) U Ue UV U[ U$ U U U UUrmr c |jr d|jd}nd}|jr d|jd}nd}|jrdt|jd}nd}|jr d|j}nd}|jrdt |j}nd}|jrd t|jd }nd}|jrd t|jd }nd}d |j d|||||||d S)Nrrrz , overload='rrz , attribute=rrrrrzFunctionProto('r) roverloadrr attributerrrrre)rrrrrrrrs rZ_function_proto_reprrsL {,dk,,, }2$-222 y1#di..111 ~34>33   W)B4CT)U)UWW   z5C OO555 {9c$+..999 jTY j j j j< j jPV jXa jcg j j jjrmprotosSequence[OperatorSetIdProto]c8d|D}t|S)Nc(i|]}|j|jS)rversion).0rs rZ z-_operator_set_protos_repr..sEEEUU\5=EEErm)repr)r opset_importss rZrrs#EEfEEEM   rm)rJrKrLrM)rXrMrJrKrLr\)rJrbrLrc)N)rnrorJrprLrq)NT)rJrKr|ror}r~rLr$)rJrKr|rorLr3)rrr|rrLr$)rrMr|rrLr3)rrrJrKr|rorr~rr~rrcrrrr~rLr\)rr3rJrKr|rorLr\)rr$rLrg)rr!rLrg)rr rLrg)rrrLrg)a __future__r__all__rhrUrBrConnxrDonnx.onnx_cpp2py_exportronnx.external_data_helperr:rAr9 onnx.onnx_pbrrrr r!r"rrrrr r r r r rrr$r%r&r(r*rr-r.r/r0r1r2r3r4r5r6r7r8onnx.onnx_operators_pbr'r)onnx.onnx_data_pbr#r+r, onnx.versionrrrrrrrrrrrr TYPE_CHECKINGcollections.abcrEr __version__rgrrwr[rarlr{r<r>r;r=r?r@loadload_from_stringsaverrrr__repr__rrmrZrs #"""""C C C L ++++++ $$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$JCBBBBBBBDDDDDDDDDD                             )((((((l" EFL$$$$MQ+++++*'+#D'+WWWWW* /FFFFF* /GGGGG,'+2 #($(#222222p'+*)@@@@8VVVV0kkkk@ ( ' - rm