From 840fc8dbca9cfcfaaa6a2f7333294c7b740ed775 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Tue, 26 Oct 2021 13:36:17 +0200 Subject: [PATCH] Add vision_encoder_decoder to models/__init__.py (#14151) * Add vision_encoder_decoder * Update _ignore_modules in get_model_modules() Co-authored-by: ydshieh --- src/transformers/models/__init__.py | 1 + utils/check_repo.py | 1 + 2 files changed, 2 insertions(+) diff --git a/src/transformers/models/__init__.py b/src/transformers/models/__init__.py index ec961e4252..9b8c5916cd 100644 --- a/src/transformers/models/__init__.py +++ b/src/transformers/models/__init__.py @@ -89,6 +89,7 @@ from . import ( t5, tapas, transfo_xl, + vision_encoder_decoder, visual_bert, vit, wav2vec2, diff --git a/utils/check_repo.py b/utils/check_repo.py index dd0a53358d..163819e415 100644 --- a/utils/check_repo.py +++ b/utils/check_repo.py @@ -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):