SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 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
2 changed files with 2 additions and 0 deletions
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, t5,
tapas, tapas,
transfo_xl, transfo_xl,
vision_encoder_decoder,
visual_bert, visual_bert,
vit, vit,
wav2vec2, wav2vec2,

1
utils/check_repo.py
View File

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