Merge pull request #288 from huggingface/gpt2

forgot to add regex to requirements.txt :(
huggingface · Feb 18, 2019 · 8f46cd1 · 8f46cd1
2 parents 0856a23 + e0855e8
commit 8f46cd1
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/pytorch_pretrained_bert/__init__.py b/pytorch_pretrained_bert/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.6.0"
+__version__ = "0.6.1"
 from .tokenization import BertTokenizer, BasicTokenizer, WordpieceTokenizer
 from .tokenization_openai import OpenAIGPTTokenizer
 from .tokenization_transfo_xl import (TransfoXLTokenizer, TransfoXLCorpus)

diff --git a/requirements.txt b/requirements.txt
@@ -5,4 +5,6 @@ tqdm
 # Accessing files from S3 directly.
 boto3
 # Used for downloading models over HTTP
-requests
+requests
+# For OpenAI GPT
+regex
diff --git a/setup.py b/setup.py
@@ -38,7 +38,7 @@
 
 setup(
     name="pytorch_pretrained_bert",
-    version="0.6.0",
+    version="0.6.1",
     author="Thomas Wolf, Victor Sanh, Tim Rault, Google AI Language Team Authors, Open AI team Authors",
     author_email="thomas@huggingface.co",
     description="PyTorch version of Google AI BERT model with script to load Google pre-trained models",
@@ -53,7 +53,8 @@
                       'numpy',
                       'boto3',
                       'requests',
-                      'tqdm'],
+                      'tqdm',
+                      'regex'],
     entry_points={
       'console_scripts': [
         "pytorch_pretrained_bert=pytorch_pretrained_bert.__main__:main",