Parcourir la source

Optimize text splitting

Lengyue il y a 1 an
Parent
commit
9f85e00327
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      tools/llama/generate.py

+ 1 - 1
tools/llama/generate.py

@@ -420,7 +420,7 @@ def split_text(text, min_length):
     curr = ""
     for char in text:
         curr += char
-        if char not in [".", ",", "!", "?"]:
+        if char not in [".", "!", "?"]:
             continue
 
         if len(curr) >= min_length: