Skip to content
Merged
Changes from all commits
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
59 changes: 0 additions & 59 deletions packages/tasks/src/pipelines.ts
Original file line number Diff line number Diff line change
Expand Up @@ -43,10 +43,6 @@ export interface PipelineData {
name: string;
subtasks?: SubTask[];
modality: Modality;
/**
* color for the tag icon.
*/
color: "blue" | "green" | "indigo" | "orange" | "red" | "yellow";
/**
* whether to hide in /models filters
*/
Expand Down Expand Up @@ -143,7 +139,6 @@ export const PIPELINE_DATA = {
},
],
modality: "nlp",
color: "orange",
},
"token-classification": {
name: "Token Classification",
Expand Down Expand Up @@ -174,12 +169,10 @@ export const PIPELINE_DATA = {
},
],
modality: "nlp",
color: "blue",
},
"table-question-answering": {
name: "Table Question Answering",
modality: "nlp",
color: "green",
},
"question-answering": {
name: "Question Answering",
Expand All @@ -198,17 +191,14 @@ export const PIPELINE_DATA = {
},
],
modality: "nlp",
color: "blue",
},
"zero-shot-classification": {
name: "Zero-Shot Classification",
modality: "nlp",
color: "yellow",
},
translation: {
name: "Translation",
modality: "nlp",
color: "green",
},
summarization: {
name: "Summarization",
Expand All @@ -223,12 +213,10 @@ export const PIPELINE_DATA = {
},
],
modality: "nlp",
color: "indigo",
},
"feature-extraction": {
name: "Feature Extraction",
modality: "nlp",
color: "red",
},
"text-generation": {
name: "Text Generation",
Expand All @@ -251,7 +239,6 @@ export const PIPELINE_DATA = {
},
],
modality: "nlp",
color: "indigo",
},
"text2text-generation": {
name: "Text2Text Generation",
Expand Down Expand Up @@ -286,7 +273,6 @@ export const PIPELINE_DATA = {
},
],
modality: "nlp",
color: "indigo",
},
"fill-mask": {
name: "Fill-Mask",
Expand All @@ -301,32 +287,26 @@ export const PIPELINE_DATA = {
},
],
modality: "nlp",
color: "red",
},
"sentence-similarity": {
name: "Sentence Similarity",
modality: "nlp",
color: "yellow",
},
"text-to-speech": {
name: "Text-to-Speech",
modality: "audio",
color: "yellow",
},
"text-to-audio": {
name: "Text-to-Audio",
modality: "audio",
color: "yellow",
},
"automatic-speech-recognition": {
name: "Automatic Speech Recognition",
modality: "audio",
color: "yellow",
},
"audio-to-audio": {
name: "Audio-to-Audio",
modality: "audio",
color: "blue",
},
"audio-classification": {
name: "Audio Classification",
Expand All @@ -353,23 +333,19 @@ export const PIPELINE_DATA = {
},
],
modality: "audio",
color: "green",
},
"audio-text-to-text": {
name: "Audio-Text-to-Text",
modality: "multimodal",
color: "red",
hideInDatasets: true,
},
"voice-activity-detection": {
name: "Voice Activity Detection",
modality: "audio",
color: "red",
},
"depth-estimation": {
name: "Depth Estimation",
modality: "cv",
color: "yellow",
},
"image-classification": {
name: "Image Classification",
Expand All @@ -384,7 +360,6 @@ export const PIPELINE_DATA = {
},
],
modality: "cv",
color: "blue",
},
"object-detection": {
name: "Object Detection",
Expand All @@ -399,7 +374,6 @@ export const PIPELINE_DATA = {
},
],
modality: "cv",
color: "yellow",
},
"image-segmentation": {
name: "Image Segmentation",
Expand All @@ -418,12 +392,10 @@ export const PIPELINE_DATA = {
},
],
modality: "cv",
color: "green",
},
"text-to-image": {
name: "Text-to-Image",
modality: "cv",
color: "yellow",
},
"image-to-text": {
name: "Image-to-Text",
Expand All @@ -434,7 +406,6 @@ export const PIPELINE_DATA = {
},
],
modality: "cv",
color: "red",
},
"image-to-image": {
name: "Image-to-Image",
Expand All @@ -453,27 +424,22 @@ export const PIPELINE_DATA = {
},
],
modality: "cv",
color: "indigo",
},
"image-to-video": {
name: "Image-to-Video",
modality: "cv",
color: "indigo",
},
"unconditional-image-generation": {
name: "Unconditional Image Generation",
modality: "cv",
color: "green",
},
"video-classification": {
name: "Video Classification",
modality: "cv",
color: "blue",
},
"reinforcement-learning": {
name: "Reinforcement Learning",
modality: "rl",
color: "red",
},
robotics: {
name: "Robotics",
Expand All @@ -488,7 +454,6 @@ export const PIPELINE_DATA = {
name: "Task Planning",
},
],
color: "blue",
},
"tabular-classification": {
name: "Tabular Classification",
Expand All @@ -503,7 +468,6 @@ export const PIPELINE_DATA = {
name: "Tabular Multi Label Classification",
},
],
color: "blue",
},
"tabular-regression": {
name: "Tabular Regression",
Expand All @@ -514,7 +478,6 @@ export const PIPELINE_DATA = {
name: "Tabular Single Column Regression",
},
],
color: "blue",
},
"tabular-to-text": {
name: "Tabular to Text",
Expand All @@ -525,13 +488,11 @@ export const PIPELINE_DATA = {
name: "RDF to text",
},
],
color: "blue",
hideInModels: true,
},
"table-to-text": {
name: "Table to Text",
modality: "nlp",
color: "blue",
hideInModels: true,
},
"multiple-choice": {
Expand All @@ -547,13 +508,11 @@ export const PIPELINE_DATA = {
},
],
modality: "nlp",
color: "blue",
hideInModels: true,
},
"text-ranking": {
name: "Text Ranking",
modality: "nlp",
color: "red",
},
"text-retrieval": {
name: "Text Retrieval",
Expand All @@ -576,7 +535,6 @@ export const PIPELINE_DATA = {
},
],
modality: "nlp",
color: "indigo",
hideInModels: true,
},
"time-series-forecasting": {
Expand All @@ -592,17 +550,14 @@ export const PIPELINE_DATA = {
name: "Multivariate Time Series Forecasting",
},
],
color: "blue",
},
"text-to-video": {
name: "Text-to-Video",
modality: "cv",
color: "green",
},
"image-text-to-text": {
name: "Image-Text-to-Text",
modality: "multimodal",
color: "red",
hideInDatasets: true,
},
"visual-question-answering": {
Expand All @@ -614,7 +569,6 @@ export const PIPELINE_DATA = {
},
],
modality: "multimodal",
color: "red",
},
"document-question-answering": {
name: "Document Question Answering",
Expand All @@ -625,48 +579,39 @@ export const PIPELINE_DATA = {
},
],
modality: "multimodal",
color: "blue",
hideInDatasets: true,
},
"zero-shot-image-classification": {
name: "Zero-Shot Image Classification",
modality: "cv",
color: "yellow",
},
"graph-ml": {
name: "Graph Machine Learning",
modality: "other",
color: "green",
},
"mask-generation": {
name: "Mask Generation",
modality: "cv",
color: "indigo",
},
"zero-shot-object-detection": {
name: "Zero-Shot Object Detection",
modality: "cv",
color: "yellow",
},
"text-to-3d": {
name: "Text-to-3D",
modality: "cv",
color: "yellow",
},
"image-to-3d": {
name: "Image-to-3D",
modality: "cv",
color: "green",
},
"image-feature-extraction": {
name: "Image Feature Extraction",
modality: "cv",
color: "indigo",
},
"video-text-to-text": {
name: "Video-Text-to-Text",
modality: "multimodal",
color: "blue",
hideInDatasets: false,
},
"keypoint-detection": {
Expand All @@ -678,25 +623,21 @@ export const PIPELINE_DATA = {
},
],
modality: "cv",
color: "red",
hideInDatasets: true,
},
"visual-document-retrieval": {
name: "Visual Document Retrieval",
modality: "multimodal",
color: "yellow",
hideInDatasets: true,
},
"any-to-any": {
name: "Any-to-Any",
modality: "multimodal",
color: "yellow",
hideInDatasets: true,
},
other: {
name: "Other",
modality: "other",
color: "blue",
hideInModels: true,
hideInDatasets: true,
},
Expand Down
Loading