Correct lstrip regex (#92)

minimaxir · Feb 26, 2021 · d3b60aa · d3b60aa
1 parent 63cdd5d
commit d3b60aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aitextgen/aitextgen.py b/aitextgen/aitextgen.py
@@ -400,7 +400,7 @@ def generate(
 
             # Handle stripping tokenization spaces w/ regex
             if lstrip:
-                gen_texts = [re.sub(r"^\W+", "", text) for text in gen_texts]
+                gen_texts = [re.sub(r"^\s+", "", text) for text in gen_texts]
 
             if not return_as_list:
                 if prompt: