Skip to content

Commit 90e39a7

Browse files
TESLER-18: Update PyTorch version compatibility to support 2.4.0+
- Updated torch constraint from >=2.2.0,<=2.3.0 to >=2.2.0,<3.0.0 - Updated torchvision constraint from >=0.17.0,<=0.18.0 to >=0.17.0,<1.0.0 - Bumped package version to 2.5.3 - Resolves user requests for PyTorch 2.4.0+ support
1 parent 787da06 commit 90e39a7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

setup.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import setuptools, os
22

33
PACKAGE_NAME = 'facenet-pytorch'
4-
VERSION = '2.5.2'
4+
VERSION = '2.5.3'
55
AUTHOR = 'Tim Esler'
66
77
DESCRIPTION = 'Pretrained Pytorch face detection and recognition models'
@@ -39,8 +39,8 @@
3939
'numpy>=1.24.0,<2.0.0',
4040
'Pillow>=10.2.0,<10.3.0',
4141
'requests>=2.0.0,<3.0.0',
42-
'torch>=2.2.0,<=2.3.0',
43-
'torchvision>=0.17.0,<=0.18.0',
42+
'torch>=2.2.0,<3.0.0',
43+
'torchvision>=0.17.0,<1.0.0',
4444
'tqdm>=4.0.0,<5.0.0',
4545
],
4646
)

0 commit comments

Comments
 (0)