Spaces:
Runtime error
Runtime error
machineuser
commited on
Commit
•
b7f922a
1
Parent(s):
2b0dab0
Sync widgets demo
Browse files
packages/tasks/src/library-to-tasks.ts
CHANGED
@@ -6,7 +6,7 @@ import type { PipelineType } from "./pipelines";
|
|
6 |
* Inference API (serverless) should be disabled for all other (library, task) pairs beyond this mapping.
|
7 |
* This mapping is partially generated automatically by "python-api-export-tasks" action in
|
8 |
* huggingface/api-inference-community repo upon merge. For transformers, the mapping is manually
|
9 |
-
* based on api-inference.
|
10 |
*/
|
11 |
export const LIBRARY_TASK_MAPPING: Partial<Record<ModelLibraryKey, PipelineType[]>> = {
|
12 |
"adapter-transformers": ["question-answering", "text-classification", "token-classification"],
|
@@ -53,12 +53,20 @@ export const LIBRARY_TASK_MAPPING: Partial<Record<ModelLibraryKey, PipelineType[
|
|
53 |
"fill-mask",
|
54 |
"image-classification",
|
55 |
"image-segmentation",
|
56 |
-
"image-to-text",
|
57 |
"image-to-image",
|
|
|
58 |
"object-detection",
|
59 |
"question-answering",
|
60 |
-
"
|
|
|
61 |
"text2text-generation",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
62 |
"visual-question-answering",
|
63 |
"zero-shot-classification",
|
64 |
"zero-shot-image-classification",
|
|
|
6 |
* Inference API (serverless) should be disabled for all other (library, task) pairs beyond this mapping.
|
7 |
* This mapping is partially generated automatically by "python-api-export-tasks" action in
|
8 |
* huggingface/api-inference-community repo upon merge. For transformers, the mapping is manually
|
9 |
+
* based on api-inference (hf_types.rs).
|
10 |
*/
|
11 |
export const LIBRARY_TASK_MAPPING: Partial<Record<ModelLibraryKey, PipelineType[]>> = {
|
12 |
"adapter-transformers": ["question-answering", "text-classification", "token-classification"],
|
|
|
53 |
"fill-mask",
|
54 |
"image-classification",
|
55 |
"image-segmentation",
|
|
|
56 |
"image-to-image",
|
57 |
+
"image-to-text",
|
58 |
"object-detection",
|
59 |
"question-answering",
|
60 |
+
"summarization",
|
61 |
+
"table-question-answering",
|
62 |
"text2text-generation",
|
63 |
+
"text-classification",
|
64 |
+
"text-generation",
|
65 |
+
"text-to-audio",
|
66 |
+
"text-to-speech",
|
67 |
+
"token-classification",
|
68 |
+
"translation",
|
69 |
+
"video-classification",
|
70 |
"visual-question-answering",
|
71 |
"zero-shot-classification",
|
72 |
"zero-shot-image-classification",
|