Skip to content

Commit f057317

Browse files
committed
Add DeiTFeatureExtractor
1 parent 76b8556 commit f057317

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/processors.js

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -280,6 +280,7 @@ export class ImageFeatureExtractor extends FeatureExtractor {
280280

281281
export class ViTFeatureExtractor extends ImageFeatureExtractor { }
282282
export class MobileViTFeatureExtractor extends ImageFeatureExtractor { }
283+
export class DeiTFeatureExtractor extends ImageFeatureExtractor { }
283284

284285
/**
285286
* Detr Feature Extractor.
@@ -1298,6 +1299,7 @@ export class AutoProcessor {
12981299
'WhisperFeatureExtractor': WhisperFeatureExtractor,
12991300
'ViTFeatureExtractor': ViTFeatureExtractor,
13001301
'MobileViTFeatureExtractor': MobileViTFeatureExtractor,
1302+
'DeiTFeatureExtractor': DeiTFeatureExtractor,
13011303
'DetrFeatureExtractor': DetrFeatureExtractor,
13021304

13031305
'SamImageProcessor': SamImageProcessor,

0 commit comments

Comments
 (0)