Add type hints for ViLT models (#18577) - SemanticDiff

Commit

3 years ago

Add type hints for ViLT models (#18577) * Add type hints for Vilt models * Add missing return type for TokenClassification class

References

#62 - Add initial DEIMv2 model implementation

#58 - Add EoMT DINOv3 model

#59 - Fix attention mask handling in EoMT-DINOv3 converter

#27720 - Add common processor tests

#32831 - [Docs] Update resources

#29969 - [SigLIP] Add fast tokenizer

#41212 - Add EoMT with DINOv3 backbone

#39821 - Support MetaCLIP 2

#33111 - [Backbone] Remove out_features everywhere

#33174 - [Zero-shot image classification pipeline] Remove tokenizer_kwargs

#19449 - [WIP] Fix weights initialization of several vision models

#18577 - Add type hints for ViLT models

Author

donelianc

donelianc

Parents

Loading