update supported models (#2813)

huggingface · Jan 31, 2025 · 37e79dc · 37e79dc
1 parent 2fdc6f4
commit 37e79dc
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 8 deletions.
diff --git a/src/huggingface_hub/inference/_providers/fal_ai.py b/src/huggingface_hub/inference/_providers/fal_ai.py
@@ -17,16 +17,16 @@
         "openai/whisper-large-v3": "fal-ai/whisper",
     },
     "text-to-image": {
-        "black-forest-labs/FLUX.1-schnell": "fal-ai/flux/schnell",
         "black-forest-labs/FLUX.1-dev": "fal-ai/flux/dev",
-        "playgroundai/playground-v2.5-1024px-aesthetic": "fal-ai/playground-v25",
+        "black-forest-labs/FLUX.1-schnell": "fal-ai/flux/schnell",
         "ByteDance/SDXL-Lightning": "fal-ai/lightning-models",
+        "fal/AuraFlow-v0.2": "fal-ai/aura-flow",
+        "Kwai-Kolors/Kolors": "fal-ai/kolors",
         "PixArt-alpha/PixArt-Sigma-XL-2-1024-MS": "fal-ai/pixart-sigma",
+        "playgroundai/playground-v2.5-1024px-aesthetic": "fal-ai/playground-v25",
         "stabilityai/stable-diffusion-3-medium": "fal-ai/stable-diffusion-v3-medium",
-        "Warlord-K/Sana-1024": "fal-ai/sana",
-        "fal/AuraFlow-v0.2": "fal-ai/aura-flow",
         "stabilityai/stable-diffusion-3.5-large": "fal-ai/stable-diffusion-v35-large",
-        "Kwai-Kolors/Kolors": "fal-ai/kolors",
+        "Warlord-K/Sana-1024": "fal-ai/sana",
     },
     "text-to-video": {
         "genmo/mochi-1-preview": "fal-ai/mochi-v1",

diff --git a/src/huggingface_hub/inference/_providers/replicate.py b/src/huggingface_hub/inference/_providers/replicate.py
@@ -23,7 +23,7 @@
         "stabilityai/stable-diffusion-xl-base-1.0": "stability-ai/sdxl:7762fd07cf82c948538e41f63f77d685e02b063e37e496e96eefd46c929f9bdc",
     },
     "text-to-speech": {
-        "OuteAI/OuteTTS-0.3-500M": "jbilcke/oute-tts:39a59319327b27327fa3095149c5a746e7f2aee18c75055c3368237a6503cd26",
+        "hexgrad/Kokoro-82M": "jaaari/kokoro-82m:f559560eb822dc509045f3921a1921234918b91739db4bf3daab2169b71c7a13",
     },
     "text-to-video": {
         "genmo/mochi-1-preview": "genmoai/mochi-1:1944af04d098ef69bed7f9d335d102e652203f268ec4aaa2d836f6217217e460",
@@ -138,7 +138,7 @@ def _prepare_payload(
         # The following payload might work only for a subset of text-to-speech Replicate models.
         payload: Dict[str, Any] = {
             "input": {
-                "inputs": inputs,
+                "text": inputs,
                 **{k: v for k, v in parameters.items() if v is not None},
             },
         }

diff --git a/src/huggingface_hub/inference/_providers/together.py b/src/huggingface_hub/inference/_providers/together.py
@@ -16,8 +16,8 @@
     "conversational": {
         "databricks/dbrx-instruct": "databricks/dbrx-instruct",
         "deepseek-ai/DeepSeek-R1": "deepseek-ai/DeepSeek-R1",
-        "deepseek-ai/DeepSeek-V3": "deepseek-ai/DeepSeek-V3",
         "deepseek-ai/deepseek-llm-67b-chat": "deepseek-ai/deepseek-llm-67b-chat",
+        "deepseek-ai/DeepSeek-V3": "deepseek-ai/DeepSeek-V3",
         "google/gemma-2-9b-it": "google/gemma-2-9b-it",
         "google/gemma-2b-it": "google/gemma-2-27b-it",
         "meta-llama/Llama-2-13b-chat-hf": "meta-llama/Llama-2-13b-chat-hf",
@@ -33,6 +33,7 @@
         "meta-llama/Meta-Llama-3.1-8B-Instruct": "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo",
         "microsoft/WizardLM-2-8x22B": "microsoft/WizardLM-2-8x22B",
         "mistralai/Mistral-7B-Instruct-v0.3": "mistralai/Mistral-7B-Instruct-v0.3",
+        "mistralai/Mistral-Small-24B-Instruct-2501": "mistralai/Mistral-Small-24B-Instruct-2501",
         "mistralai/Mixtral-8x22B-Instruct-v0.1": "mistralai/Mixtral-8x22B-Instruct-v0.1",
         "mistralai/Mixtral-8x7B-Instruct-v0.1": "mistralai/Mixtral-8x7B-Instruct-v0.1",
         "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO": "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",