rotary needs to be done with full precision to be safe

This commit is contained in:
lucidrains
2024-05-11 08:04:14 -07:00
parent bca88e9039
commit 90be7233a3
2 changed files with 4 additions and 1 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.8',
version = '1.6.9',
license='MIT',
description = 'Vision Transformer (ViT) - Pytorch',
long_description=long_description,