Bump dependencies

dbanys · Jan 17, 2021 · a842bc3 · a842bc3
1 parent fdb6778
commit a842bc3
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/aitextgen/aitextgen.py b/aitextgen/aitextgen.py
@@ -324,7 +324,7 @@ def generate(
             schema_return = getattr(self.model.config, "schema_return")
             schema_tokens_enc = self.tokenizer(text=schema_tokens)["input_ids"]
 
-            nonalphanum_pattern = re.compile(r"[\W_]+")
+            nonalphanum_pattern = re.compile(r"[\W_]+", re.UNICODE)
 
             outputs = outputs.tolist()
             gen_texts = []

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
-transformers>=4.0.0
+transformers>=4.2.1
 fire>=0.3.0
-pytorch-lightning>=1.0.8
+pytorch-lightning>=1.1.4
 tokenizers>=0.9.4
 torch>=1.6.0
diff --git a/setup.py b/setup.py
@@ -17,9 +17,9 @@
     python_requires=">=3.6",
     include_package_data=True,
     install_requires=[
-        "transformers>=4.0.0",
+        "transformers>=4.2.1",
         "fire>=0.3.0",
-        "pytorch-lightning>=1.0.8",
+        "pytorch-lightning>=1.1.4",
         "tokenizers>=0.9.4",
         "torch>=1.6.0",
     ],