TiUSrSSKJr SS/rSSKrSSKJr SSKrSSKJ r J r J r SSKJ r SS KJr S S S jjrSSS jjrg)zLoad and save ONNX models.) annotationsloadsaveN)Callable)_core _protocolsserde) external_data)zipc[R"XSS9n[R"U5n[R R U5n[R"URU5 U$)zLoad an ONNX model from a file. Args: path: The path to the ONNX file. format: The format of the file (e.g. protobuf, textproto, json, etc.). If None, the format is inferred from the file extension. Returns: The loaded model. F)formatload_external_data) onnxrr deserialize_modelospathdirname_external_data set_base_dirgraph)rr protomodelbase_dirs J/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx_ir/_io.pyrrsQ IIde DE  # #E *Ewwt$H X6 LcUGb [RRU5(a[SUS35e[RR U5n/nUR 5H,nUR URR55 M. UV s/sHoRPM n n [R"UUUUUS9n[R"U5n [R"XUS9 [!XzSS9H upXl M g[R"U5n [R"XUS9 gs sn f![!XzSS9H upXl M f=f)aSave an ONNX model to a file. The model remains unchanged after the call. If any existing external tensor references the provided ``external_data`` path, it will be invalidated after the external data is overwritten. To obtain a valid model, use :func:`load` to load the newly saved model, or provide a different external data path that is not currently referenced by any tensors in the model. .. tip:: A simple progress bar can be implemented by passing a callback function as the following:: import onnx_ir as ir import tqdm with tqdm.tqdm() as pbar: total_set = False def callback(tensor: ir.TensorProtocol, metadata: ir.external_data.CallbackInfo) -> None: nonlocal total_set if not total_set: pbar.total = metadata.total total_set = True pbar.update() pbar.set_description(f"Saving {tensor.name} ({tensor.dtype}, {tensor.shape}) at offset {metadata.offset}") ir.save( ..., callback=callback, ) Args: model: The model to save. path: The path to save the model to. E.g. "model.onnx". format: The format of the file (e.g. ``protobuf``, ``textproto``, ``json``, etc.). If None, the format is inferred from the file extension. external_data: The relative path to save external data to. When specified, all initializers in the model will be converted to external data and saved to the specified directory. If None, all tensors will be saved unmodified. That is, if a tensor in the model is already external, it will be saved with the same external information; if the tensor is not external, it will be serialized in the ONNX Proto message. size_threshold_bytes: Save to external data if the tensor size in bytes is larger than this threshold. Effective only when ``external_data`` is set. callback: A callback function that is called for each tensor that is saved to external data for debugging or logging purposes. Raises: ValueError: If the external data path is an absolute path. NzDThe external data path must be relative to the ONNX file path, not 'z'.)size_threshold_bytescallback)r T)strict)rrisabs ValueErrorrgraphsextend initializersvalues const_valuerunload_from_modelr serialize_modelrrr )rrr r rrrinitialized_valuesrvtensorsr initializertensors rrr)sDx 77== ' 'VWdVeegh 77??4(13\\^E  % %e&8&8&?&?&A B$+==*(++=t'T# *0'(UsD>5AEE)N)rstr | os.PathLiker str | Nonereturn _core.Model)NNN)rr1rr.r r/r zstr | os.PathLike | NonerintrzOCallable[[_protocols.TensorProtocol, _external_data.CallbackInfo], None] | Noner0None)__doc__ __future__r__all__rtypingrronnx_irrrr r ronnx_ir._polyfillr rrrrr<s!" 6   ,,3!2.2 # \. \. \. \., \.  \.  \. \.r