From fb88a12c4c42087215a2f4a39f73888592a23d03 Mon Sep 17 00:00:00 2001 From: MrPleom-sirius <100464767+MrPleom-sirius@users.noreply.github.com> Date: Thu, 17 Aug 2023 17:03:30 +0800 Subject: [PATCH] Update setup.py --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 281c337..d6691f2 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', 'pytorch-pretrained-bert==0.6.2', 'tensorflow', ],