From 6e07361a2f15de7925a3cdefc91d32390b7489c9 Mon Sep 17 00:00:00 2001
From: Max Woolf <max@minimaxir.com>
Date: Sun, 16 May 2021 19:37:09 -0700
Subject: [PATCH] version bumps

---
 requirements.txt | 2 +-
 setup.py         | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 93f2c86..f3efead 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 transformers>=4.5.1
 fire>=0.3.0
-pytorch-lightning>=1.2.7
+pytorch-lightning>=1.3.1
 torch>=1.6.0
\ No newline at end of file
diff --git a/setup.py b/setup.py
index f4402d8..d6ec36a 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 setup(
     name="aitextgen",
     packages=["aitextgen"],  # this must be the same as the name above
-    version="0.5.1",
+    version="0.5.2",
     description="A robust Python tool for text-based AI training and generation using GPT-2.",
     long_description=open("README.md", "r", encoding="utf-8").read(),
     long_description_content_type="text/markdown",
@@ -19,7 +19,7 @@
     install_requires=[
         "transformers>=4.5.1",
         "fire>=0.3.0",
-        "pytorch-lightning>=1.2.7",
+        "pytorch-lightning>=1.3.1",
         "torch>=1.6.0",
     ],
 )