瀏覽代碼

fix io bug

Lengyue 1 年之前
父節點
當前提交
acfbf17b8e
共有 1 個文件被更改,包括 1 次插入5 次删除
  1. 1 5
      fish_speech/utils/file.py

+ 1 - 5
fish_speech/utils/file.py

@@ -44,11 +44,7 @@ def list_files(
     if not path.exists():
         raise FileNotFoundError(f"Directory {path} does not exist.")
 
-    # files = [Path(f) for f in glob(str(path / "**/*"), recursive=recursive)]
-    files = [file for ext in extensions for file in directory.glob(f"**/*{ext}")]
-
-    # if extensions is not None:
-    #    files = [f for f in files if f.suffix in extensions]
+    files = [file for ext in extensions for file in path.iglob(f"**/*{ext}")]
 
     if sort:
         files = natsorted(files)