From 60bd8111db7bcefb94c27c3f9f74af26fa415cdb Mon Sep 17 00:00:00 2001 From: Howard Lander Date: Mon, 6 Dec 2021 13:39:16 -0500 Subject: [PATCH] updated setup.py to require a version of torch that is compatible with python3.9 --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 281c337..7d4bc98 100644 --- a/setup.py +++ b/setup.py @@ -15,7 +15,7 @@ url="https://github.com/overfitter/biobert_embedding", download_url="https://github.com/overfitter/biobert_embedding/archive/v0.1.2.tar.gz", install_requires=[ - 'torch==1.2.0', + 'torch==1.7.1', 'pytorch-pretrained-bert==0.6.2', 'tensorflow', ],