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
- "text-generation",
 
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",