optimum
71ed6465 - `vit` + `wav2vec2` support (#470)

Commit

2 years ago

`vit` + `wav2vec2` support (#470) * `vit` support - refactor tests * add `wav2vec2` * add `hubert` * fix tests - fix docstring + refactor tests * Update optimum/bettertransformer/transformation.py Co-authored-by: fxmarty <9808326+fxmarty@users.noreply.github.com> * remove duplicate tests * add `hubert` on doc * add all supported models * add last model * fix `hubert` test * Update tests/bettertransformer/test_bettertransformer_vision.py Co-authored-by: fxmarty <9808326+fxmarty@users.noreply.github.com> * fix tests add docstring Co-authored-by: fxmarty <9808326+fxmarty@users.noreply.github.com>

References

#470 - `vit` + `wav2vec2` support

Author

younesbelkada

Parents

48c6e945

Files9

docs/source/bettertransformer
- overview.mdx
optimum/bettertransformer
- models
  - __init__.py
  - encoder_models.py
- transformation.py
tests
- benchmark
  - benchmark_bettertransformer_vit.py
- bettertransformer
  - test_bettertransformer_audio.py
  - test_bettertransformer_encoder.py
  - test_bettertransformer_vision.py
  - testing_bettertransformer_utils.py

optimum 71ed6465 - `vit` + `wav2vec2` support (#470)

optimum
71ed6465 - `vit` + `wav2vec2` support (#470)