Skip to content

Commit

Permalink
allow all torch 1.6.x versions incl 1.6.0+cu101 etc
Browse files Browse the repository at this point in the history
Former-commit-id: 0c4046a
Former-commit-id: 17990e32fb6571e3acfc922d2da09a2899299a19
  • Loading branch information
tholor committed Sep 21, 2020
1 parent 867cc46 commit 77ebb98
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ setuptools
wheel
# PyTorch
--find-links=https://download.pytorch.org/whl/torch_stable.html
torch>=1.5.1,<=1.6.0
torch>1.5,<1.7
# progress bars in model download and training scripts
tqdm
# Accessing files from S3 directly.
Expand Down
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ def get_dependency_links(filename):

setup(
name="farm",
version="0.4.8",
version="0.4.9",
author="Malte Pietsch, Timo Moeller, Branden Chan, Tanay Soni",
author_email="[email protected]",
description="Toolkit for finetuning and evaluating transformer based language models",
Expand All @@ -55,7 +55,7 @@ def get_dependency_links(filename):
keywords="BERT NLP deep learning language-model transformer qa question-answering transfer-learning",
license="Apache",
url="https://gitlab.com/deepset-ai/ml/lm/farm",
download_url="https://github.com/deepset-ai/FARM/archive/0.4.8.tar.gz",
download_url="https://github.com/deepset-ai/FARM/archive/0.4.9.tar.gz",
packages=find_packages(exclude=["*.tests", "*.tests.*", "tests.*", "tests"]),
dependency_links=dependency_links,
install_requires=parsed_requirements,
Expand Down

0 comments on commit 77ebb98

Please sign in to comment.