Bläddra i källkod

fix: package requirements allow minor versions of torch

NexPB 4 år sedan
förälder
incheckning
5c1201d454
1 ändrade filer med 2 tillägg och 2 borttagningar
  1. 2 2
      requirements.txt

+ 2 - 2
requirements.txt

@@ -2,8 +2,8 @@ flask==1.1.2
 numpy==1.19.4
 numpy==1.19.4
 pillow==8.0.1
 pillow==8.0.1
 scikit-image==0.17.2
 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
 waitress==1.4.4
 tqdm==4.51.0
 tqdm==4.51.0
 requests==2.24.0
 requests==2.24.0