From dc30f2c14658747f6185e475b95952edb27d12ea Mon Sep 17 00:00:00 2001 From: Matt Date: Tue, 16 Apr 2024 14:25:47 +0100 Subject: [PATCH] Re-add removed idefics2 line --- src/transformers/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py index c744cfe35e8e0b..8f340c50bb159f 100644 --- 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 (