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
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 details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file setfitonnx-0.1.0.tar.gz.
File metadata
- Download URL: setfitonnx-0.1.0.tar.gz
- Upload date:
- Size: 4.4 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.9.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
70705f41b52578e72d720a955cd1c9fab7a54371bf4a3e29eec29e00fd0a21e5
|
|
| MD5 |
b71aa9f24e4c6229872dae9950fa3e72
|
|
| BLAKE2b-256 |
14fd2c6b6347f38bf0b5dbfc8978e9d4a1f0ef96e13090aa55ab06afc1595b5d
|
File details
Details for the file setfitonnx-0.1.0-py3-none-any.whl.
File metadata
- Download URL: setfitonnx-0.1.0-py3-none-any.whl
- Upload date:
- Size: 5.0 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.9.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
5a492526c62f0bb623c5b509592791ddad92ad63d84286ea9fc2447de2cbf862
|
|
| MD5 |
a283c468bf902bf8bf42a6d867226f7c
|
|
| BLAKE2b-256 |
afdeab2b1f4a11e833959a118191ef4c203aefe1bb266f12772673ed86173e77
|
