diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py
index c8e237b4e8..ac852cedf4 100644
--- a/optimum/exporters/tasks.py
+++ b/optimum/exporters/tasks.py
@@ -222,6 +222,7 @@ class TasksManager:
             "multiple-choice": "TFAutoModelForMultipleChoice",
             "object-detection": "TFAutoModelForObjectDetection",
             "question-answering": "TFAutoModelForQuestionAnswering",
+            "image-feature-extraction": "TFAutoModel",
             "image-segmentation": "TFAutoModelForImageSegmentation",
             "masked-im": "TFAutoModelForMaskedImageModeling",
             "semantic-segmentation": "TFAutoModelForSemanticSegmentation",
diff --git a/optimum/exporters/tflite/base.py b/optimum/exporters/tflite/base.py
index 3df230c33b..71ee7ca953 100644
--- a/optimum/exporters/tflite/base.py
+++ b/optimum/exporters/tflite/base.py
@@ -160,6 +160,7 @@ class TFLiteConfig(ExportConfig, ABC):
         "text-generation": ["logits"],
         "feature-extraction": ["last_hidden_state"],
         "image-classification": ["logits"],
+        "image-feature-extraction": ["last_hidden_state"],
         "image-segmentation": ["logits", "pred_boxes", "pred_masks"],
         "masked-im": ["logits"],
         "fill-mask": ["logits"],