Skip to content
Merged
Show file tree
Hide file tree
Changes from 4 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
6 changes: 6 additions & 0 deletions packages/tasks/src/pipelines.ts
Original file line number Diff line number Diff line change
Expand Up @@ -693,6 +693,12 @@ export const PIPELINE_DATA = {
color: "yellow",
hideInDatasets: true,
},
"video-to-video": {
name: "Video-to-Video",
modality: "cv",
color: "blue",
hideInDatasets: true,
},
other: {
name: "Other",
modality: "other",
Expand Down
3 changes: 2 additions & 1 deletion packages/tasks/src/tasks/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,6 @@ export type {
} from "./zero-shot-object-detection/inference.js";

import type { ModelLibraryKey } from "../model-libraries.js";

/**
* Model libraries compatible with each ML task
*/
Expand Down Expand Up @@ -181,6 +180,7 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
"image-to-3d": ["diffusers"],
"any-to-any": ["transformers"],
"visual-document-retrieval": ["transformers"],
"video-to-video": ["diffusers"],
};

/**
Expand Down Expand Up @@ -250,6 +250,7 @@ export const TASKS_DATA: Record<PipelineType, TaskData | undefined> = {
translation: getData("translation", translation),
"unconditional-image-generation": getData("unconditional-image-generation", unconditionalImageGeneration),
"video-text-to-text": getData("video-text-to-text", videoTextToText),
"video-to-video": getData("video-to-video", placeholder),
"visual-question-answering": getData("visual-question-answering", visualQuestionAnswering),
"voice-activity-detection": undefined,
"zero-shot-classification": getData("zero-shot-classification", zeroShotClassification),
Expand Down
Loading