machineuser commited on
Commit
de47a68
1 Parent(s): 096c59d

Sync widgets demo

Browse files
Files changed (1) hide show
  1. packages/tasks/src/const.ts +7 -7
packages/tasks/src/const.ts CHANGED
@@ -16,10 +16,10 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
16
  "graph-ml": ["transformers"],
17
  "image-classification": ["keras", "timm", "transformers", "transformers.js"],
18
  "image-segmentation": ["transformers", "transformers.js"],
19
- "image-to-image": [],
20
  "image-to-text": ["transformers.js"],
21
  "image-to-video": ["diffusers"],
22
- "video-classification": [],
23
  "mask-generation": ["transformers"],
24
  "multiple-choice": ["transformers"],
25
  "object-detection": ["transformers", "transformers.js"],
@@ -37,10 +37,10 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
37
  "text-classification": ["adapter-transformers", "spacy", "transformers", "transformers.js"],
38
  "text-generation": ["transformers", "transformers.js"],
39
  "text-retrieval": [],
40
- "text-to-image": [],
41
  "text-to-speech": ["espnet", "tensorflowtts", "transformers"],
42
  "text-to-audio": ["transformers"],
43
- "text-to-video": [],
44
  "text2text-generation": ["transformers", "transformers.js"],
45
  "time-series-forecasting": [],
46
  "token-classification": [
@@ -53,10 +53,10 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
53
  "transformers.js",
54
  ],
55
  translation: ["transformers", "transformers.js"],
56
- "unconditional-image-generation": [],
57
- "visual-question-answering": [],
58
  "voice-activity-detection": [],
59
  "zero-shot-classification": ["transformers", "transformers.js"],
60
- "zero-shot-image-classification": ["transformers.js"],
61
  "zero-shot-object-detection": ["transformers"],
62
  };
 
16
  "graph-ml": ["transformers"],
17
  "image-classification": ["keras", "timm", "transformers", "transformers.js"],
18
  "image-segmentation": ["transformers", "transformers.js"],
19
+ "image-to-image": ["diffusers"],
20
  "image-to-text": ["transformers.js"],
21
  "image-to-video": ["diffusers"],
22
+ "video-classification": ["transformers"],
23
  "mask-generation": ["transformers"],
24
  "multiple-choice": ["transformers"],
25
  "object-detection": ["transformers", "transformers.js"],
 
37
  "text-classification": ["adapter-transformers", "spacy", "transformers", "transformers.js"],
38
  "text-generation": ["transformers", "transformers.js"],
39
  "text-retrieval": [],
40
+ "text-to-image": ["diffusers"],
41
  "text-to-speech": ["espnet", "tensorflowtts", "transformers"],
42
  "text-to-audio": ["transformers"],
43
+ "text-to-video": ["diffusers"],
44
  "text2text-generation": ["transformers", "transformers.js"],
45
  "time-series-forecasting": [],
46
  "token-classification": [
 
53
  "transformers.js",
54
  ],
55
  translation: ["transformers", "transformers.js"],
56
+ "unconditional-image-generation": ["diffusers"],
57
+ "visual-question-answering": ["transformers"],
58
  "voice-activity-detection": [],
59
  "zero-shot-classification": ["transformers", "transformers.js"],
60
+ "zero-shot-image-classification": ["transformers", "transformers.js"],
61
  "zero-shot-object-detection": ["transformers"],
62
  };