offer 1d versions, in light of https://arxiv.org/abs/2211.14730

2025-12-30 08:02:29 +00:00 · 2022-12-01 10:31:05 -08:00
parent 59c8948c6a
commit 2f87c0cf8f
3 changed files with 260 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
 setup(
  name = 'vit-pytorch',
  packages = find_packages(exclude=['examples']),
-  version = '0.38.1',
+  version = '0.39.1',
  license='MIT',
  description = 'Vision Transformer (ViT) - Pytorch',
  long_description_content_type = 'text/markdown',
@@ -16,7 +16,7 @@ setup(
    'image recognition'
  ],
  install_requires=[
-    'einops>=0.4.1',
+    'einops>=0.6.0',
    'torch>=1.10',
    'torchvision'
  ],