diff --git a/setup.py b/setup.py
index 2d359a5489..f5cadca56e 100644
--- a/setup.py
+++ b/setup.py
@@ -155,7 +155,7 @@ def package_files(prefix: str, directory: str, extension: str):
 ]
 
 extra_deps['vit'] = [
-    'vit_pytorch==0.27',
+    'vit_pytorch==0.35.8',
 ]
 
 extra_deps['timm'] = [