Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 33 additions & 3 deletions docs/inference-providers/providers/hf-inference.md
Original file line number Diff line number Diff line change
Expand Up @@ -79,6 +79,16 @@ Find out more about Feature Extraction [here](../tasks/feature_extraction).
/>


### Fill Mask

Find out more about Fill Mask [here](../tasks/fill_mask).

<InferenceSnippet
pipeline=fill-mask
providersMapping={ {"hf-inference":{"modelId":"FacebookAI/roberta-base","providerModelId":"FacebookAI/roberta-base"} } }
/>


### Image Classification

Find out more about Image Classification [here](../tasks/image_classification).
Expand All @@ -89,6 +99,16 @@ Find out more about Image Classification [here](../tasks/image_classification).
/>


### Image Segmentation

Find out more about Image Segmentation [here](../tasks/image_segmentation).

<InferenceSnippet
pipeline=image-segmentation
providersMapping={ {"hf-inference":{"modelId":"jonathandinu/face-parsing","providerModelId":"jonathandinu/face-parsing"} } }
/>


### Object Detection

Find out more about Object Detection [here](../tasks/object_detection).
Expand All @@ -105,7 +125,7 @@ Find out more about Question Answering [here](../tasks/question_answering).

<InferenceSnippet
pipeline=question-answering
providersMapping={ {"hf-inference":{"modelId":"deepset/bert-large-uncased-whole-word-masking-squad2","providerModelId":"deepset/bert-large-uncased-whole-word-masking-squad2"} } }
providersMapping={ {"hf-inference":{"modelId":"deepset/roberta-large-squad2","providerModelId":"deepset/roberta-large-squad2"} } }
/>


Expand All @@ -125,7 +145,7 @@ Find out more about Text Classification [here](../tasks/text_classification).

<InferenceSnippet
pipeline=text-classification
providersMapping={ {"hf-inference":{"modelId":"j-hartmann/emotion-english-distilroberta-base","providerModelId":"j-hartmann/emotion-english-distilroberta-base"} } }
providersMapping={ {"hf-inference":{"modelId":"SamLowe/roberta-base-go_emotions","providerModelId":"SamLowe/roberta-base-go_emotions"} } }
/>


Expand All @@ -149,13 +169,23 @@ Find out more about Text To Image [here](../tasks/text_to_image).
/>


### Token Classification

Find out more about Token Classification [here](../tasks/token_classification).

<InferenceSnippet
pipeline=token-classification
providersMapping={ {"hf-inference":{"modelId":"dslim/bert-base-NER","providerModelId":"dslim/bert-base-NER"} } }
/>


### Translation

Find out more about Translation [here](../tasks/translation).

<InferenceSnippet
pipeline=translation
providersMapping={ {"hf-inference":{"modelId":"facebook/nllb-200-distilled-600M","providerModelId":"facebook/nllb-200-distilled-600M"} } }
providersMapping={ {"hf-inference":{"modelId":"facebook/mbart-large-50-many-to-many-mmt","providerModelId":"facebook/mbart-large-50-many-to-many-mmt"} } }
/>


Expand Down
1 change: 0 additions & 1 deletion docs/inference-providers/tasks/chat-completion.md
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,6 @@ This is a subtask of [`text-generation`](https://huggingface.co/docs/inference-p
#### Conversational Large Language Models (LLMs)

- [google/gemma-2-2b-it](https://huggingface.co/google/gemma-2-2b-it): A text-generation model trained to follow instructions.
- [deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B](https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B): Smaller variant of one of the most powerful models.
- [meta-llama/Meta-Llama-3.1-8B-Instruct](https://huggingface.co/meta-llama/Meta-Llama-3.1-8B-Instruct): Very powerful text generation model trained to follow instructions.
- [microsoft/phi-4](https://huggingface.co/microsoft/phi-4): Powerful text generation model by Microsoft.
- [Qwen/Qwen2.5-Coder-32B-Instruct](https://huggingface.co/Qwen/Qwen2.5-Coder-32B-Instruct): Text generation model used to write code.
Expand Down
5 changes: 4 additions & 1 deletion docs/inference-providers/tasks/fill-mask.md
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,10 @@ Explore all available models and find the one that suits you best [here](https:/
### Using the API


No snippet available for this task.
<InferenceSnippet
pipeline=fill-mask
providersMapping={ {"hf-inference":{"modelId":"FacebookAI/roberta-base","providerModelId":"FacebookAI/roberta-base"}} }
/>



Expand Down
5 changes: 4 additions & 1 deletion docs/inference-providers/tasks/image-segmentation.md
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,10 @@ Explore all available models and find the one that suits you best [here](https:/
### Using the API


No snippet available for this task.
<InferenceSnippet
pipeline=image-segmentation
providersMapping={ {"hf-inference":{"modelId":"jonathandinu/face-parsing","providerModelId":"jonathandinu/face-parsing"}} }
/>



Expand Down
2 changes: 1 addition & 1 deletion docs/inference-providers/tasks/question-answering.md
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ Explore all available models and find the one that suits you best [here](https:/

<InferenceSnippet
pipeline=question-answering
providersMapping={ {"hf-inference":{"modelId":"deepset/bert-large-uncased-whole-word-masking-squad2","providerModelId":"deepset/bert-large-uncased-whole-word-masking-squad2"}} }
providersMapping={ {"hf-inference":{"modelId":"deepset/roberta-large-squad2","providerModelId":"deepset/roberta-large-squad2"}} }
/>


Expand Down
3 changes: 2 additions & 1 deletion docs/inference-providers/tasks/text-classification.md
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ For more details about the `text-classification` task, check out its [dedicated

### Recommended models

- [papluca/xlm-roberta-base-language-detection](https://huggingface.co/papluca/xlm-roberta-base-language-detection): A model that can classify languages.

Explore all available models and find the one that suits you best [here](https://huggingface.co/models?inference=warm&pipeline_tag=text-classification&sort=trending).

Expand All @@ -32,7 +33,7 @@ Explore all available models and find the one that suits you best [here](https:/

<InferenceSnippet
pipeline=text-classification
providersMapping={ {"hf-inference":{"modelId":"j-hartmann/emotion-english-distilroberta-base","providerModelId":"j-hartmann/emotion-english-distilroberta-base"}} }
providersMapping={ {"hf-inference":{"modelId":"SamLowe/roberta-base-go_emotions","providerModelId":"SamLowe/roberta-base-go_emotions"}} }
/>


Expand Down
1 change: 0 additions & 1 deletion docs/inference-providers/tasks/text-generation.md
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,6 @@ For more details about the `text-generation` task, check out its [dedicated page
### Recommended models

- [google/gemma-2-2b-it](https://huggingface.co/google/gemma-2-2b-it): A text-generation model trained to follow instructions.
- [deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B](https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B): Smaller variant of one of the most powerful models.
- [meta-llama/Meta-Llama-3.1-8B-Instruct](https://huggingface.co/meta-llama/Meta-Llama-3.1-8B-Instruct): Very powerful text generation model trained to follow instructions.
- [microsoft/phi-4](https://huggingface.co/microsoft/phi-4): Powerful text generation model by Microsoft.
- [Qwen/Qwen2.5-Coder-32B-Instruct](https://huggingface.co/Qwen/Qwen2.5-Coder-32B-Instruct): Text generation model used to write code.
Expand Down
1 change: 1 addition & 0 deletions docs/inference-providers/tasks/text-to-image.md
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ For more details about the `text-to-image` task, check out its [dedicated page](
### Recommended models

- [black-forest-labs/FLUX.1-dev](https://huggingface.co/black-forest-labs/FLUX.1-dev): One of the most powerful image generation models that can generate realistic outputs.
- [latent-consistency/lcm-lora-sdxl](https://huggingface.co/latent-consistency/lcm-lora-sdxl): A powerful yet fast image generation model.
- [Kwai-Kolors/Kolors](https://huggingface.co/Kwai-Kolors/Kolors): Text-to-image model for photorealistic generation.
- [stabilityai/stable-diffusion-3-medium-diffusers](https://huggingface.co/stabilityai/stable-diffusion-3-medium-diffusers): A powerful text-to-image model.

Expand Down
6 changes: 5 additions & 1 deletion docs/inference-providers/tasks/token-classification.md
Original file line number Diff line number Diff line change
Expand Up @@ -24,13 +24,17 @@ For more details about the `token-classification` task, check out its [dedicated

### Recommended models

- [dslim/bert-base-NER](https://huggingface.co/dslim/bert-base-NER): A robust performance model to identify people, locations, organizations and names of miscellaneous entities.

Explore all available models and find the one that suits you best [here](https://huggingface.co/models?inference=warm&pipeline_tag=token-classification&sort=trending).

### Using the API


No snippet available for this task.
<InferenceSnippet
pipeline=token-classification
providersMapping={ {"hf-inference":{"modelId":"dslim/bert-base-NER","providerModelId":"dslim/bert-base-NER"}} }
/>



Expand Down
2 changes: 1 addition & 1 deletion docs/inference-providers/tasks/translation.md
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ Explore all available models and find the one that suits you best [here](https:/

<InferenceSnippet
pipeline=translation
providersMapping={ {"hf-inference":{"modelId":"facebook/nllb-200-distilled-600M","providerModelId":"facebook/nllb-200-distilled-600M"}} }
providersMapping={ {"hf-inference":{"modelId":"facebook/mbart-large-50-many-to-many-mmt","providerModelId":"facebook/mbart-large-50-many-to-many-mmt"}} }
/>


Expand Down
4 changes: 2 additions & 2 deletions scripts/inference-providers/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,8 @@
"author": "",
"license": "ISC",
"dependencies": {
"@huggingface/inference": "^3.8.1",
"@huggingface/tasks": "^0.18.9",
"@huggingface/inference": "^3.9.2",
"@huggingface/tasks": "^0.18.11",
"@types/node": "^22.5.0",
"handlebars": "^4.7.8",
"node": "^20.17.0",
Expand Down
22 changes: 11 additions & 11 deletions scripts/inference-providers/pnpm-lock.yaml

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading