소스 검색

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)