Merge branch 'main' of github.com:JonathanFly/bark into main

JonathanFly · Jun 8, 2023 · 6da63d1 · 6da63d1
2 parents 7a14f52 + 774def4
commit 6da63d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bark_infinity/text_processing.py b/bark_infinity/text_processing.py
@@ -276,7 +276,7 @@ def split_text(text: str, split_type: Optional[str] = None, split_type_quantity
             #print(split_type_to_function[split_type](current_segment, split_type=split_type_value_type, split_type_quantity=1, split_type_string=split_type_string))
             split_type_quantity_found = len(split_type_to_function[split_type_value_type](current_segment, split_type=split_type_value_type, split_type_quantity=1, split_type_string=split_type_string))
             #print(f"I see {split_type_quantity_found} {split_type_value_type} in {current_segment}")
-            if split_type_quantity_found >= split_type_quantity:
+            if split_type_quantity_found >= int(split_type_quantity):
                 final_segmented_text.append(current_segment)
                 split_type_quantity_found = 0
                 current_segment = ''