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

* 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

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):