Add vision models to doc tests (#15905)

* Add vision models to doc tests

* Apply suggestions from code review

* Add more models

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
This commit is contained in:
NielsRogge
2022-03-03 19:46:31 +01:00
committed by GitHub
parent 742273a52a
commit 9251427c38
5 changed files with 50 additions and 16 deletions

View File

@@ -9,7 +9,16 @@ src/transformers/models/sew/modeling_sew.py
src/transformers/models/sew_d/modeling_sew_d.py
src/transformers/models/speech_to_text_2/modeling_speech_to_text_2.py
src/transformers/models/speech_to_text/modeling_speech_to_text.py
src/transformers/models/speech_encoder_decoder/modeling_speech_enocder_decoder.py
src/transformers/models/speech_encoder_decoder/modeling_speech_encoder_decoder.py
src/transformers/models/data2vec/modeling_data2vec_audio.py
src/transformers/models/vit/modeling_vit.py
src/transformers/models/beit/modeling_beit.py
src/transformers/models/deit/modeling_deit.py
src/transformers/models/swin/modeling_swin.py
src/transformers/models/convnext/modeling_convnext.py
src/transformers/models/poolformer/modeling_poolformer.py
src/transformers/models/vit_mae/modeling_vit_mae.py
src/transformers/models/segformer/modeling_segformer.py
src/transformers/models/vision_encoder_decoder/modeling_vision_encoder_decoder.py
docs/source/quicktour.mdx
docs/source/task_summary.mdx
docs/source/task_summary.mdx