APIsΒΆ

onnx_safetensors.save_model

Save an ONNX model to a file with external data 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.load_file_as_external_data

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

onnx_safetensors.replace_tensors

Replace all tensors in an ONNX model with external data from a 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.