Fix Errors in `setfit-onnx-optimum` Notebook (#496) · huggingface/setfit@82df763 · GitHub

Commit

Fix Errors in setfit-onnx-optimum Notebook (#496)

Browse files

* ONNX export bug workaround; use a faster setfit model (with SetFitHead); Run on A100 80GB

* minor

* Update setfit-onnx-optimum.ipynb

Loading branch information

danielkorat committed Mar 11, 2024

1 parent 793ca23 commit 82df763