Metadata-Version: 2.1
Name: setfitonnx
Version: 0.1.0
Summary: Package to pack setfit model to onnx format
Home-page: https://github.com/vimal-b/setfitonnx
License: MIT
Keywords: setfit,onnx,sentence-transformers,onnxruntime
Author: Vimal Babu
Author-email: personal.vimal@gmail.com
Requires-Python: >=3.9,<4.0
Classifier: Environment :: Console
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.9
Classifier: Topic :: Software Development :: Documentation
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Classifier: Topic :: Software Development :: Quality Assurance
Requires-Dist: joblib (>=1.2.0,<2.0.0)
Requires-Dist: loguru (>=0.6.0,<0.7.0)
Requires-Dist: onnx (>=1.12.0,<2.0.0)
Requires-Dist: onnxruntime (>=1.13.1,<2.0.0)
Requires-Dist: sclblonnx (>=0.1.14,<0.2.0)
Requires-Dist: setfit (>=0.4.1,<0.5.0)
Requires-Dist: skl2onnx (>=1.13,<2.0)
Project-URL: Repository, https://github.com/vimal-b/setfitonnx
Description-Content-Type: text/markdown

# SetFitONNX
## Export the setfit model to ONNX format


## Features

- Mean Pooling Layer to ONNX Graph
- Single ONNX Model combining both the sentence transformer & classification head

## Installation
```sh
pip install setfitonnx
```
## How to 
```python
from setfitonnx import convert_onnx

# Setfit Model directory
model_path = "/home/mysetfit-model"
# ONNX Output directory
output_dir = "/home/setfit-onnx-model"
# Convert to ONNX
convert_onnx(model_path=model_path,output_dir=output_dir)

```

## License

MIT
