add a card for organization
Browse files
README.md
CHANGED
|
@@ -7,4 +7,97 @@ sdk: static
|
|
| 7 |
pinned: false
|
| 8 |
---
|
| 9 |
|
| 10 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
pinned: false
|
| 8 |
---
|
| 9 |
|
| 10 |
+
Welcome to **ONNX Working Group**! 🤗
|
| 11 |
+
|
| 12 |
+
We are looking for contributing to HuggingFace's ONNX implementation for all available models on the HF's hub.
|
| 13 |
+
|
| 14 |
+
There is already a lot of architectures implemented for converting PyTorch models to ONNX, but we need more! We need
|
| 15 |
+
them all!
|
| 16 |
+
|
| 17 |
+
Feel free to join me in this adventure! You can reach me directly on Discord: `ChainYo#3610`
|
| 18 |
+
|
| 19 |
+
Here is a list of all the models that need to receive ONNX support:
|
| 20 |
+
|
| 21 |
+
- ✅ Albert
|
| 22 |
+
- ✅ BART
|
| 23 |
+
- BeiT
|
| 24 |
+
- ✅ BERT
|
| 25 |
+
- BigBird
|
| 26 |
+
- BigBirdPegasus
|
| 27 |
+
- Blenderbot
|
| 28 |
+
- BlenderbotSmall
|
| 29 |
+
- CLIP
|
| 30 |
+
- CTRL
|
| 31 |
+
- ✅ CamemBERT
|
| 32 |
+
- Canine
|
| 33 |
+
- ConvBert
|
| 34 |
+
- ✅ Data2Vec
|
| 35 |
+
- Deberta
|
| 36 |
+
- DeiT
|
| 37 |
+
- ✅ Distilbert
|
| 38 |
+
- ✅ Electra
|
| 39 |
+
- FNet
|
| 40 |
+
- FSMT
|
| 41 |
+
- Flaubert
|
| 42 |
+
- Funnel
|
| 43 |
+
- GPT2
|
| 44 |
+
- GPTJ
|
| 45 |
+
- ✅ GPTNeo
|
| 46 |
+
- Hubert
|
| 47 |
+
- ✅ I-Bert
|
| 48 |
+
- ImageGPT
|
| 49 |
+
- LED
|
| 50 |
+
- ✅ LayoutLM
|
| 51 |
+
- LayoutLMv2
|
| 52 |
+
- Longformer
|
| 53 |
+
- Luke
|
| 54 |
+
- Lxmert
|
| 55 |
+
- ✅ M2M100
|
| 56 |
+
- ✅ mBart
|
| 57 |
+
- MPNet
|
| 58 |
+
- MT5
|
| 59 |
+
- ✅ Marian
|
| 60 |
+
- MegatronBert
|
| 61 |
+
- MobileBert
|
| 62 |
+
- Nystromformer
|
| 63 |
+
- ✅ OpenAIGPT-2
|
| 64 |
+
- ✅ PLBart
|
| 65 |
+
- Pegasus
|
| 66 |
+
- Perceiver
|
| 67 |
+
- PoolFormer
|
| 68 |
+
- ProphetNet
|
| 69 |
+
- rag
|
| 70 |
+
- Reformer
|
| 71 |
+
- RemBert
|
| 72 |
+
- RoFormer
|
| 73 |
+
- ✅ RoBERTa
|
| 74 |
+
- SEWD
|
| 75 |
+
- Segformer
|
| 76 |
+
- Splinter
|
| 77 |
+
- SqueezeBERT
|
| 78 |
+
- Swin
|
| 79 |
+
- ✅ T5
|
| 80 |
+
- TransfoXL
|
| 81 |
+
- UniSpeech
|
| 82 |
+
- UniSpeechSat
|
| 83 |
+
- ViT
|
| 84 |
+
- Vilt
|
| 85 |
+
- VisualBERT
|
| 86 |
+
- Wav2Vec2
|
| 87 |
+
- WavLM
|
| 88 |
+
- XGLM
|
| 89 |
+
- XLM
|
| 90 |
+
- XLMProphetNet
|
| 91 |
+
- ✅ XLM-RoBERTa
|
| 92 |
+
- ✅ XLM-Roberta-XL
|
| 93 |
+
- XLNet
|
| 94 |
+
- Yoso
|
| 95 |
+
|
| 96 |
+
If there is a green checkmark next to a model, it means that the model is already supported by ONNX. If there is nothing
|
| 97 |
+
next to a model, it means that the model is not yet supported by ONNX and thus we need to add support for it.
|
| 98 |
+
|
| 99 |
+
If you need some help about implementing an unsupported model, here is a [guide from HF's](https://huggingface.co/docs/transformers/serialization#exporting-a-model-for-an-unsupported-architecture) documentation.
|
| 100 |
+
|
| 101 |
+
If you want an example of implementation, I did one for [CamemBERT](https://github.com/huggingface/transformers/pull/14059) some months ago.
|
| 102 |
+
|
| 103 |
+
Have fun coding!
|