Browse Source

fix: package requirements allow minor versions of torch

NexPB 4 years ago
parent
commit
5c1201d454
1 changed files with 2 additions and 2 deletions
  1. 2 2
      requirements.txt

+ 2 - 2
requirements.txt

@@ -2,8 +2,8 @@ flask==1.1.2
 numpy==1.19.4
 pillow==8.0.1
 scikit-image==0.17.2
-torch==1.7.0
-torchvision==0.8.1
+torch~=1.7.0
+torchvision~=0.8.1
 waitress==1.4.4
 tqdm==4.51.0
 requests==2.24.0