Add Segment Anything Model (SAM)

docs/source/en/model_doc/sam.mdx

@@ -38,9 +38,6 @@ The original code can be found [here](<INSERT LINK TO GITHUB REPO HERE>).
 
 [[autodoc]] SamVisionConfig
 
-## SamQFormerConfig
-
-[[autodoc]] SamQFormerConfig
 
 ## SamProcessor
 

src/transformers/__init__.py

@@ -200,7 +200,6 @@
         "SAM_PRETRAINED_CONFIG_ARCHIVE_MAP",
         "SamConfig",
         "SamProcessor",
-        "SamQFormerConfig",
         "SamVisionConfig",
     ],
     "models.bloom": ["BLOOM_PRETRAINED_CONFIG_ARCHIVE_MAP", "BloomConfig"],
@@ -1196,11 +1195,7 @@
     _import_structure["models.sam"].extend(
         [
             "SAM_PRETRAINED_MODEL_ARCHIVE_LIST",
-            "SamForConditionalGeneration",
-            "SamModel",
-            "SamPreTrainedModel",
-            "SamQFormerModel",
-            "SamVisionModel",
+            "SamForImageSegmentation",
         ]
     )
     _import_structure["models.bloom"].extend(
@@ -3905,7 +3900,6 @@
         SAM_PRETRAINED_CONFIG_ARCHIVE_MAP,
         SamConfig,
         SamProcessor,
-        SamQFormerConfig,
         SamVisionConfig,
     )
     from .models.bloom import BLOOM_PRETRAINED_CONFIG_ARCHIVE_MAP, BloomConfig
@@ -4790,11 +4784,7 @@
         )
         from .models.sam import (
             SAM_PRETRAINED_MODEL_ARCHIVE_LIST,
-            SamForConditionalGeneration,
-            SamModel,
-            SamPreTrainedModel,
-            SamQFormerModel,
-            SamVisionModel,
+            SamForImageSegmentation,
         )
         from .models.bloom import (
             BLOOM_PRETRAINED_MODEL_ARCHIVE_LIST,

src/transformers/models/sam/__init__.py

@@ -20,7 +20,6 @@
     "configuration_sam": [
         "SAM_PRETRAINED_CONFIG_ARCHIVE_MAP",
         "SamConfig",
-        "SamQFormerConfig",
         "SamVisionConfig",
     ],
     "processing_sam": ["SamProcessor"],
@@ -34,18 +33,13 @@
 else:
     _import_structure["modeling_sam"] = [
         "SAM_PRETRAINED_MODEL_ARCHIVE_LIST",
-        "SamModel",
-        "SamQFormerModel",
-        "SamPreTrainedModel",
-        "SamForConditionalGeneration",
-        "SamVisionModel",
+        "SamForImageSegmentation",
     ]
 
 if TYPE_CHECKING:
     from .configuration_sam import (
         SAM_PRETRAINED_CONFIG_ARCHIVE_MAP,
         SamConfig,
-        SamQFormerConfig,
         SamVisionConfig,
     )
     from .processing_sam import SamProcessor
@@ -58,11 +52,7 @@
     else:
         from .modeling_sam import (
             SAM_PRETRAINED_MODEL_ARCHIVE_LIST,
-            SamForConditionalGeneration,
-            SamModel,
-            SamPreTrainedModel,
-            SamQFormerModel,
-            SamVisionModel,
+            SamForImageSegmentation
         )
 
 else: