Unverified Commit 840fc8db authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

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



* Add vision_encoder_decoder

* Update _ignore_modules in get_model_modules()
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent e248e9b0
...@@ -89,6 +89,7 @@ from . import ( ...@@ -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,
......
...@@ -167,6 +167,7 @@ def get_model_modules(): ...@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment