Re-add removed idefics2 line

huggingface · Apr 16, 2024 · dc30f2c · dc30f2c
1 parent 8ea3258
commit dc30f2c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -5433,6 +5433,7 @@
     from .models.hubert import HUBERT_PRETRAINED_CONFIG_ARCHIVE_MAP, HubertConfig
     from .models.ibert import IBERT_PRETRAINED_CONFIG_ARCHIVE_MAP, IBertConfig
     from .models.idefics import IDEFICS_PRETRAINED_CONFIG_ARCHIVE_MAP, IdeficsConfig
+    from .models.idefics2 import Idefics2Config
     from .models.imagegpt import IMAGEGPT_PRETRAINED_CONFIG_ARCHIVE_MAP, ImageGPTConfig
     from .models.informer import INFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP, InformerConfig
     from .models.instructblip import (