APIsΒΆ

onnx_safetensors.save_model

Save an ONNX model to a file with external data in a safetensors file.

onnx_safetensors.load_file_as_external_data

Load weights from safetensors file and use them as external data for the ONNX model.

onnx_safetensors.save_safetensors_model

Save an ONNX model to a safetensors file.

onnx_safetensors.extract_safetensors_model

Unpack an ONNX model stored in a safetensors file.

onnx_safetensors.load

Load external data into ONNX model from safetensors bytes.

onnx_safetensors.load_file

Load external data into ONNX model from a safetensors file.

onnx_safetensors.replace_tensors

Replace weights in an ONNX model with those of the same names found in the safetensors file.

onnx_safetensors.save

Save all tensors in an ONNX model to a safetensors object serialized as bytes.

onnx_safetensors.save_file

Save all tensors in an ONNX model to a safetensors file.