Add DeiTFeatureExtractor

huggingface · Aug 26, 2023 · f057317 · f057317
1 parent 76b8556
commit f057317
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/processors.js b/src/processors.js
@@ -280,6 +280,7 @@ export class ImageFeatureExtractor extends FeatureExtractor {
 
 export class ViTFeatureExtractor extends ImageFeatureExtractor { }
 export class MobileViTFeatureExtractor extends ImageFeatureExtractor { }
+export class DeiTFeatureExtractor extends ImageFeatureExtractor { }
 
 /**
  * Detr Feature Extractor.
@@ -1298,6 +1299,7 @@ export class AutoProcessor {
         'WhisperFeatureExtractor': WhisperFeatureExtractor,
         'ViTFeatureExtractor': ViTFeatureExtractor,
         'MobileViTFeatureExtractor': MobileViTFeatureExtractor,
+        'DeiTFeatureExtractor': DeiTFeatureExtractor,
         'DetrFeatureExtractor': DetrFeatureExtractor,
 
         'SamImageProcessor': SamImageProcessor,