|
|
--- |
|
|
title: Convert to ONNX |
|
|
emoji: ☯ |
|
|
colorFrom: indigo |
|
|
colorTo: yellow |
|
|
sdk: docker |
|
|
app_port: 8501 |
|
|
pinned: true |
|
|
license: apache-2.0 |
|
|
short_description: Convert a Hugging Face model to ONNX format |
|
|
--- |
|
|
|
|
|
# Convert to ONNX |
|
|
|
|
|
## Overview |
|
|
|
|
|
This project provides a Streamlit application that facilitates the conversion of Hugging Face models to ONNX format, downloading, converting, and uploading models to Hugging Face. |
|
|
|
|
|
## Docker Compose usage |
|
|
|
|
|
### 1. Prepare environment variables |
|
|
|
|
|
Copy the provided template and fill in your Hugging Face write token: |
|
|
|
|
|
```bash |
|
|
cp .env.example .env |
|
|
``` |
|
|
|
|
|
### 2. Start the application |
|
|
|
|
|
```bash |
|
|
docker compose up |
|
|
``` |
|
|
|
|
|
Access the interface at `http://localhost:8501`. |
|
|
|
|
|
Enter a Hugging Face model ID (e.g., `EleutherAI/pythia-14m`). |
|
|
|
|
|
After a successful conversion, the ONNX export is available under `{username}/{model-name}-ONNX` unless you opt into reusing the same repository. |
|
|
|
|
|
To stop the service, press `Ctrl+C` (or run `docker compose down`). Add `-d` to run detached. |
|
|
|
|
|
## Direct Docker usage (optional) |
|
|
|
|
|
If you prefer not to use Docker Compose, you can still build and run manually: |
|
|
|
|
|
```bash |
|
|
docker build -t convert-to-onnx . |
|
|
|
|
|
docker run --rm \ |
|
|
-p 8501:8501 \ |
|
|
-e HF_TOKEN="your_write_token" \ |
|
|
convert-to-onnx |
|
|
``` |
|
|
|
|
|
## Development |
|
|
|
|
|
### Contributing |
|
|
|
|
|
1. Fork the repository |
|
|
2. Create a feature branch |
|
|
3. Implement changes with tests |
|
|
4. Submit a pull request |
|
|
|
|
|
## Troubleshooting |
|
|
|
|
|
Common issues and solutions: |
|
|
|
|
|
- **Authentication Errors**: Verify your Hugging Face credentials |
|
|
- **Conversion Failures**: Check model compatibility and available disk space |
|
|
- **Upload Issues**: Ensure stable internet connection and valid permissions |
|
|
|
|
|
## License |
|
|
|
|
|
[Apache 2.0 License](license.txt) |
|
|
|
