Adding optimum option for PredictEngine#492

Merged

michaelfeil merged 9 commits intomichaelfeil:mainfrom wwymak:onnx-predict-engine

Dec 10, 2024

+144

Commits on Dec 6, 2024

adding optimum onnx option for classification
wendy mak
committed
removing a few unneeded bits from optimum.py
wendy mak
committed

Commits on Dec 8, 2024

adding missing required args from optimum pipeline to make sure it gives the same output as torch implementation
wendy mak
committed
adding unit test
wendy mak
committed
Merge remote-tracking branch 'origin/main' into onnx-predict-engine
wendy mak
committed
remove a few stray things that are not needed
wendy mak
committed
fixing code styles
wendy mak
committed
minor linting fix
wendy mak
committed

Commits on Dec 9, 2024

fix failing test
wendy mak
committed