Skip to content

Commit 71495e2

Browse files
Add video-to-video (#1469)
--------- Co-authored-by: Vaibhavs10 <vaibhavs10@gmail.com>
1 parent 697bf84 commit 71495e2

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

packages/tasks/src/pipelines.ts

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -632,6 +632,11 @@ export const PIPELINE_DATA = {
632632
name: "Any-to-Any",
633633
modality: "multimodal",
634634
},
635+
"video-to-video": {
636+
name: "Video-to-Video",
637+
modality: "cv",
638+
hideInDatasets: true,
639+
},
635640
other: {
636641
name: "Other",
637642
modality: "other",

packages/tasks/src/tasks/index.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,6 @@ export type {
113113
} from "./zero-shot-object-detection/inference.js";
114114

115115
import type { ModelLibraryKey } from "../model-libraries.js";
116-
117116
/**
118117
* Model libraries compatible with each ML task
119118
*/
@@ -181,6 +180,7 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
181180
"image-to-3d": ["diffusers"],
182181
"any-to-any": ["transformers"],
183182
"visual-document-retrieval": ["transformers"],
183+
"video-to-video": ["diffusers"],
184184
};
185185

186186
/**
@@ -250,6 +250,7 @@ export const TASKS_DATA: Record<PipelineType, TaskData | undefined> = {
250250
translation: getData("translation", translation),
251251
"unconditional-image-generation": getData("unconditional-image-generation", unconditionalImageGeneration),
252252
"video-text-to-text": getData("video-text-to-text", videoTextToText),
253+
"video-to-video": getData("video-to-video", placeholder),
253254
"visual-question-answering": getData("visual-question-answering", visualQuestionAnswering),
254255
"voice-activity-detection": undefined,
255256
"zero-shot-classification": getData("zero-shot-classification", zeroShotClassification),

0 commit comments

Comments
 (0)