dnrops
/
transformers.js
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add `DeiTFeatureExtractor`

This commit is contained in:
Joshua Lochner 2023-08-26 23:54:27 +02:00
parent 76b8556110
commit f0573175fd
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/processors.js
View File

@ -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,