Skip to content

Commit 3aec3d2

Browse files
committed
add snippetGenerator
1 parent 47d5221 commit 3aec3d2

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

packages/inference/src/snippets/getInferenceSnippets.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -393,7 +393,9 @@ const snippets: Partial<
393393
"fill-mask": snippetGenerator("basic"),
394394
"image-classification": snippetGenerator("basicImage"),
395395
"image-segmentation": snippetGenerator("basicImage"),
396+
"image-text-to-image": snippetGenerator("imageToImage", prepareImageToImageInput),
396397
"image-text-to-text": snippetGenerator("conversational"),
398+
"image-text-to-video": snippetGenerator("imageToVideo", prepareImageToImageInput),
397399
"image-to-image": snippetGenerator("imageToImage", prepareImageToImageInput),
398400
"image-to-text": snippetGenerator("basicImage"),
399401
"image-to-video": snippetGenerator("imageToVideo", prepareImageToImageInput),

0 commit comments

Comments
 (0)