optimum
https://github.com/huggingface/optimum
Python
🚀 Accelerate training and inference of 🤗 Transformers and 🤗 Diffusers with easy to use hardware optimization tools
Triage Issues!
When you volunteer to triage issues, you'll receive an email each day with a link to an open issue that needs help in this project. You'll also receive instructions on how to triage issues.
Triage Docs!
Receive a documented method or class from your favorite GitHub repos in your inbox every day. If you're really pro, receive undocumented methods or classes and supercharge your commit history.
Python not yet supported2 Subscribers
Add a CodeTriage badge to optimum
Help out
- Issues
- GPTQ kernel inference not compatible with some models
- Add ONNX export support for IJepa
- Allow sending directly torch/tf/np dtype for generating dummy inputs
- Expose more params in `main_export` for customized ONNX export
- Add ONNX export support for ViTPose
- TFJS support model.json to ONNX conversion
- 🐛 Fix model kwargs handling in pipeline function
- Add support for Musicgen Melody in the ONNX export
- Add visualbert onnx suport
- Update "past_sequence_length + 1" shape dimension name to "past_sequence_length + sequence_length"
- Docs
- Python not yet supported