Add image-text-to-text to libraries-tasks to enable pipeline snippets (…

…#1126) We recently merged an image-text-to-text pipeline so enabling this Co-authored-by: vb <[email protected]>
huggingface · Jan 28, 2025 · 7508458 · 7508458
1 parent d43b6a9
commit 7508458
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/packages/tasks/src/library-to-tasks.ts b/packages/tasks/src/library-to-tasks.ts
@@ -55,6 +55,8 @@ export const LIBRARY_TASK_MAPPING: Partial<Record<ModelLibraryKey, PipelineType[
 		"image-segmentation",
 		"image-to-image",
 		"image-to-text",
+		"image-text-to-text",
+		"mask-generation",
 		"object-detection",
 		"question-answering",
 		"summarization",