From ce5217763848fa172f87ded595d77af872e0d125 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Tue, 11 Dec 2018 12:12:08 +0100 Subject: [PATCH] added version in __init__.py --- pytorch_pretrained_bert/__init__.py | 1 + requirements.txt | 5 ++--- setup.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/pytorch_pretrained_bert/__init__.py b/pytorch_pretrained_bert/__init__.py index e1ecabf31d..416fe25e69 100644 --- a/pytorch_pretrained_bert/__init__.py +++ b/pytorch_pretrained_bert/__init__.py @@ -1,3 +1,4 @@ +__version__ = 0.4.0 from .tokenization import BertTokenizer, BasicTokenizer, WordpieceTokenizer from .modeling import (BertConfig, BertModel, BertForPreTraining, BertForMaskedLM, BertForNextSentencePrediction, diff --git a/requirements.txt b/requirements.txt index e9a3640a9b..f37f11cc54 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,5 @@ -# This installs Pytorch for CUDA 8 only. If you are using a newer version, -# please visit http://pytorch.org/ and install the relevant version. -torch>=0.4.1,<0.5.0 +# PyTorch +torch>=0.4.1 # progress bars in model download and training scripts tqdm # Accessing files from S3 directly. diff --git a/setup.py b/setup.py index fc793b53e6..21ca97294d 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ from setuptools import find_packages, setup setup( name="pytorch_pretrained_bert", - version="0.3.0", + version="0.4.0", author="Thomas Wolf, Victor Sanh, Tim Rault, Google AI Language Team Authors", author_email="thomas@huggingface.co", description="PyTorch version of Google AI BERT model with script to load Google pre-trained models",