Skip to content

Commit 0a690a1

Browse files
authored
Add image-feature-extraction task to transformers and timm lib->task mapping (#1120)
There's no pipeline snippet for models w/ the image-feature-extraction task, probably because this mapping is missing? cc @pcuenca re slack comment
1 parent e288dbe commit 0a690a1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

packages/tasks/src/library-to-tasks.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ export const LIBRARY_TASK_MAPPING: Partial<Record<ModelLibraryKey, PipelineType[
4343
"text2text-generation",
4444
],
4545
stanza: ["token-classification"],
46-
timm: ["image-classification"],
46+
timm: ["image-classification", "image-feature-extraction"],
4747
transformers: [
4848
"audio-classification",
4949
"automatic-speech-recognition",
@@ -52,6 +52,7 @@ export const LIBRARY_TASK_MAPPING: Partial<Record<ModelLibraryKey, PipelineType[
5252
"feature-extraction",
5353
"fill-mask",
5454
"image-classification",
55+
"image-feature-extraction",
5556
"image-segmentation",
5657
"image-to-image",
5758
"image-to-text",

0 commit comments

Comments
 (0)