Przeglądaj źródła

Optimize text splitting

Lengyue 1 rok temu
rodzic
commit
9f85e00327
1 zmienionych plików z 1 dodań i 1 usunięć
  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: