Function safe_tensors_loader::convert_onnx

pub fn convert_onnx(input: &Path, output: &Path) -> Result<(), SafeTensorError>

Convert an ONNX model to a SafeTensor weights.

Arguments

• input: Path to the ONNX model.
• output: Path to the SafeTensor weights.

Examples

convert_onnx(Path::new("tests/diffuser.onnx"), Path::new("tests/diffuser.safetensors")).unwrap();