Package to pack setfit model to onnx format
Project description
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
pip install setfitonnx
How to
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
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
setfitonnx-0.1.0.tar.gz
(4.4 kB
view hashes)
Built Distribution
Close
Hashes for setfitonnx-0.1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 5a492526c62f0bb623c5b509592791ddad92ad63d84286ea9fc2447de2cbf862 |
|
MD5 | a283c468bf902bf8bf42a6d867226f7c |
|
BLAKE2b-256 | afdeab2b1f4a11e833959a118191ef4c203aefe1bb266f12772673ed86173e77 |