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

1
.github/workflows/python-test.yml vendored
View File

@@ -28,6 +28,7 @@ jobs:
python -m pip install --upgrade pip python -m pip install --upgrade pip
python -m pip install pytest python -m pip install pytest
python -m pip install wheel python -m pip install wheel
python -m pip install torch==2.4.0 torchvision==0.19.0 --index-url https://download.pytorch.org/whl/cpu
if [ -f requirements.txt ]; then pip install -r requirements.txt; fi if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
- name: Test with pytest - name: Test with pytest
run: | run: |

4
setup.py
View File

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