some models only return embeddings with some kwarg on forward

This commit is contained in:
lucidrains
2025-07-27 08:46:43 -07:00
parent b46233c3d6
commit e05cd6d8b8
2 changed files with 4 additions and 3 deletions

View File

@@ -6,7 +6,7 @@ with open('README.md') as f:
setup(
name = 'vit-pytorch',
packages = find_packages(exclude=['examples']),
version = '1.11.2',
version = '1.11.3',
license='MIT',
description = 'Vision Transformer (ViT) - Pytorch',
long_description = long_description,