[Python snippets] map snippetConversationalWithImage

mishig25 · mishig25 · commit 1bb5b31131c6 · 2024-10-02T11:01:21.000+02:00
diff --git a/packages/tasks/src/snippets/python.ts b/packages/tasks/src/snippets/python.ts
@@ -155,6 +155,7 @@ export const pythonSnippets: Partial<Record<PipelineType, (model: ModelDataMinim
 	"feature-extraction": snippetBasic,
 	"text-generation": snippetBasic,
 	"text2text-generation": snippetBasic,
+	"image-text-to-text": snippetConversationalWithImage,
 	"fill-mask": snippetBasic,
 	"sentence-similarity": snippetBasic,
 	"automatic-speech-recognition": snippetFile,
