You must be logged in to use this space

🤗 Optimum Neuron Model Exporter 🏎️ (WIP)

This Space allows you to automatically export 🤗 transformers models hosted on the Hugging Face Hub to AWS Neuron-optimized format for Inferentia/Trainium acceleration. It opens a PR on the target model, and it is up to the owner of the original model to merge the PR to allow people to leverage Neuron optimization!

Features:

  • Automatically opens PR with Neuron-optimized model
  • Preserves original model weights
  • Adds proper tags to model card

Requirements:

  • Model must be compatible with Optimum Neuron
  • User must be logged in with write token
{}
Task (auto could infer task from model)