ZF/vit-pytorch
1
0
Fork 0
mirror of https://github.com/lucidrains/vit-pytorch.git synced 2025-12-30 08:02:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix tests

Browse Source 
fix tests
This commit is contained in:
Phil Wang
2024-08-20 15:35:04 -07:00
committed by GitHub
parent 73199ab486
commit bed48b5912
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@@ -29,8 +29,8 @@ setup(
],
tests_require=[
'pytest',
'torch==1.12.1',
'torchvision==0.13.1'
'torch==2.4.0',
'torchvision==0.19.0'
],
classifiers=[
'Development Status :: 4 - Beta',