added version in __init__.py

This commit is contained in:
thomwolf
2018-12-11 12:12:08 +01:00
parent b13abfa9fe
commit ed3b62cd3b
3 changed files with 4 additions and 4 deletions

View File

@@ -1,3 +1,4 @@
__version__ = 0.4.0
from .tokenization import BertTokenizer, BasicTokenizer, WordpieceTokenizer from .tokenization import BertTokenizer, BasicTokenizer, WordpieceTokenizer
from .modeling import (BertConfig, BertModel, BertForPreTraining, from .modeling import (BertConfig, BertModel, BertForPreTraining,
BertForMaskedLM, BertForNextSentencePrediction, BertForMaskedLM, BertForNextSentencePrediction,

View File

@@ -1,6 +1,5 @@
# This installs Pytorch for CUDA 8 only. If you are using a newer version, # PyTorch
# please visit http://pytorch.org/ and install the relevant version. torch>=0.4.1
torch>=0.4.1,<0.5.0
# progress bars in model download and training scripts # progress bars in model download and training scripts
tqdm tqdm
# Accessing files from S3 directly. # Accessing files from S3 directly.

View File

@@ -2,7 +2,7 @@ from setuptools import find_packages, setup
setup( setup(
name="pytorch_pretrained_bert", 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="Thomas Wolf, Victor Sanh, Tim Rault, Google AI Language Team Authors",
author_email="thomas@huggingface.co", author_email="thomas@huggingface.co",
description="PyTorch version of Google AI BERT model with script to load Google pre-trained models", description="PyTorch version of Google AI BERT model with script to load Google pre-trained models",