riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 10:12:23 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Add vision_encoder_decoder to models/__init__.py (#14151)

Browse Source 
* Add vision_encoder_decoder

* Update _ignore_modules in get_model_modules()

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2021-10-26 13:36:17 +02:00 committed by GitHub
parent e248e9b042
commit 840fc8dbca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/models/__init__.py
View File

@ -89,6 +89,7 @@ from . import (
t5,
tapas,
transfo_xl,
vision_encoder_decoder,
visual_bert,
vit,
wav2vec2,

1
utils/check_repo.py
View File

@ -167,6 +167,7 @@ def get_model_modules():
"modeling_tf_pytorch_utils",
"modeling_tf_utils",
"modeling_tf_transfo_xl_utilities",
"modeling_vision_encoder_decoder",
]
modules = []
for model in dir(transformers.models):