Jelajahi Sumber

Use natsort for better data packing

Lengyue 2 tahun lalu
induk
melakukan
74ec1db166
2 mengubah file dengan 4 tambahan dan 2 penghapusan
  1. 2 1
      fish_speech/utils/file.py
  2. 2 1
      pyproject.toml

+ 2 - 1
fish_speech/utils/file.py

@@ -4,6 +4,7 @@ from pathlib import Path
 from typing import Union
 
 from loguru import logger
+from natsort import natsorted
 
 AUDIO_EXTENSIONS = {
     ".mp3",
@@ -49,7 +50,7 @@ def list_files(
         files = [f for f in files if f.suffix in extensions]
 
     if sort:
-        files = sorted(files)
+        files = natsorted(files)
 
     return files
 

+ 2 - 1
pyproject.toml

@@ -34,7 +34,8 @@ dependencies = [
     "zibai-server>=0.9.0",
     "loguru>=0.6.0",
     "WeTextProcessing>=0.1.10",
-    "loralib>=0.1.2"
+    "loralib>=0.1.2",
+    "natsort>=8.4.0"
 ]
 
 [project.optional-dependencies]