Ver Fonte

fix io bug

Lengyue há 1 ano atrás
pai
commit
acfbf17b8e
1 ficheiros alterados com 1 adições e 5 exclusões
  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)