allow for qk norm to be turned off for na vit nested tensor

This commit is contained in:
lucidrains
2024-11-20 10:59:22 -08:00
parent f6d7287b6b
commit 24196a3e8a
3 changed files with 15 additions and 13 deletions

View File

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