瀏覽代碼

Merge pull request #69 from NexPB/master

fix: package requirements to allow for minor versions of torch
Daniel Gatis 4 年之前
父節點
當前提交
8e12fa3556
共有 1 個文件被更改,包括 2 次插入2 次删除
  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