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:
@@ -89,6 +89,7 @@ from . import (
|
||||
t5,
|
||||
tapas,
|
||||
transfo_xl,
|
||||
vision_encoder_decoder,
|
||||
visual_bert,
|
||||
vit,
|
||||
wav2vec2,
|
||||
|
||||
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user