diff --git a/setup.py b/setup.py
index 4db2dd9..d9a73e3 100644
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,7 @@
 
 setup(
     name='textgen',
-    version='1.0.1',
+    version='1.0.2',
     description='Text Generation Model',
     long_description=readme,
     long_description_content_type='text/markdown',
diff --git a/textgen/__init__.py b/textgen/__init__.py
index f153ce8..d77116c 100644
--- a/textgen/__init__.py
+++ b/textgen/__init__.py
@@ -4,7 +4,7 @@
 @description: 
 """
 
-__version__ = '1.0.1'
+__version__ = '1.0.2'
 
 from textgen.augment.text_augment import TextAugment