fix t2t vit having two layernorms, and make final layernorm in distillation wrapper configurable, default to False for vit

This commit is contained in:
lucidrains
2024-06-11 15:12:53 -07:00
parent 90be7233a3
commit e3256d77cd
3 changed files with 17 additions and 14 deletions

View File

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