Sfoglia il codice sorgente

Optimize text splitting

Lengyue 1 anno fa
parent
commit
9f85e00327
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  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: