diff --git a/packages/tasks/src/pipelines.ts b/packages/tasks/src/pipelines.ts index 0595f2ed86..5b651c3237 100644 --- a/packages/tasks/src/pipelines.ts +++ b/packages/tasks/src/pipelines.ts @@ -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", diff --git a/packages/tasks/src/tasks/index.ts b/packages/tasks/src/tasks/index.ts index 19c25ccfce..7bc1568816 100644 --- a/packages/tasks/src/tasks/index.ts +++ b/packages/tasks/src/tasks/index.ts @@ -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 */ @@ -181,6 +180,7 @@ export const TASKS_MODEL_LIBRARIES: Record = { "image-to-3d": ["diffusers"], "any-to-any": ["transformers"], "visual-document-retrieval": ["transformers"], + "video-to-video": ["diffusers"], }; /** @@ -250,6 +250,7 @@ export const TASKS_DATA: Record = { 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),