• NielsRogge's avatar
    Add Vision Transformer and ViTFeatureExtractor (#10950) · 30677dc7
    NielsRogge authored
    
    
    * Squash all commits into one
    
    * Update ViTFeatureExtractor to use image_utils instead of torchvision
    
    * Remove torchvision and add Pillow
    
    * Small docs improvement
    
    * Address most comments by @sgugger
    
    * Fix tests
    
    * Clean up conversion script
    
    * Pooler first draft
    
    * Fix quality
    
    * Improve conversion script
    
    * Make style and quality
    
    * Make fix-copies
    
    * Minor docs improvements
    
    * Should use fix-copies instead of manual handling
    
    * Revert "Should use fix-copies instead of manual handling"
    
    This reverts commit fd4e591bce4496d41406425c82606a8fdaf8a50b.
    
    * Place ViT in alphabetical order
    Co-authored-by: default avatarLysandre <lysandre.debut@reseau.eseo.fr>
    Co-authored-by: default avatarSylvain Gugger <35901082+sgugger@users.noreply.github.com>
    30677dc7
image_utils.py 6.18 KB