Skip to content

Commit caa4c34

Browse files
authored
1 parent a1e8e5b commit caa4c34

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

packages/tasks/src/library-to-tasks.ts

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ import type { PipelineType } from "./pipelines";
66
* Inference API (serverless) should be disabled for all other (library, task) pairs beyond this mapping.
77
* This mapping is partially generated automatically by "python-api-export-tasks" action in
88
* huggingface/api-inference-community repo upon merge. For transformers, the mapping is manually
9-
* based on api-inference.
9+
* based on api-inference (hf_types.rs).
1010
*/
1111
export const LIBRARY_TASK_MAPPING: Partial<Record<ModelLibraryKey, PipelineType[]>> = {
1212
"adapter-transformers": ["question-answering", "text-classification", "token-classification"],
@@ -53,12 +53,20 @@ export const LIBRARY_TASK_MAPPING: Partial<Record<ModelLibraryKey, PipelineType[
5353
"fill-mask",
5454
"image-classification",
5555
"image-segmentation",
56-
"image-to-text",
5756
"image-to-image",
57+
"image-to-text",
5858
"object-detection",
5959
"question-answering",
60-
"text-generation",
60+
"summarization",
61+
"table-question-answering",
6162
"text2text-generation",
63+
"text-classification",
64+
"text-generation",
65+
"text-to-audio",
66+
"text-to-speech",
67+
"token-classification",
68+
"translation",
69+
"video-classification",
6270
"visual-question-answering",
6371
"zero-shot-classification",
6472
"zero-shot-image-classification",

0 commit comments

Comments
 (0)