Fix 'sentence-transformers/all-MiniLM-L6-v2' doesn't support task 'feature-extraction' (#2968)

Author: Wauplin

src/huggingface_hub/inference/_providers/hf_inference.py

@@ -151,6 +151,14 @@ def _check_supported_task(model: str, task: str) -> None:
             return  # Only conversational allowed if tagged as conversational
         raise ValueError("Non-conversational image-text-to-text task is not supported.")
 
+    if (
+        task in ("feature-extraction", "sentence-similarity")
+        and pipeline_tag in ("feature-extraction", "sentence-similarity")
+        and task in tags
+    ):
+        # feature-extraction and sentence-similarity are interchangeable for HF Inference
+        return
+
     # For all other tasks, just check pipeline tag
     if pipeline_tag != task:
         raise ValueError(

tests/test_inference_providers.py

@@ -585,6 +585,21 @@ def test_prepare_payload_as_dict_conversational(self, mapped_model, parameters,
                 "conversational",
                 True,
             ),
+            # Feature-extraction / sentence-similarity are interchangeable for HF Inference
+            (
+                "sentence-similarity",
+                ["tag1", "feature-extraction", "sentence-similarity"],
+                "feature-extraction",
+                False,
+            ),
+            (
+                "feature-extraction",
+                ["tag1", "feature-extraction", "sentence-similarity"],
+                "sentence-similarity",
+                False,
+            ),
+            # if pipeline_tag is not feature-extraction or sentence-similarity, raise
+            ("text-generation", ["tag1", "feature-extraction", "sentence-similarity"], "sentence-similarity", True),
             # Other tasks
             (
                 "audio-classification",