From fc289b7f7b69a7fa77431f1e85c10a1bccd24b2d Mon Sep 17 00:00:00 2001 From: Ray Tien Date: Sun, 12 Jan 2025 12:50:18 +0800 Subject: [PATCH] Fix: Update incorrect import path for load_filelist function (#818) * Modify import path. * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> --- tools/llama/build_dataset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tools/llama/build_dataset.py b/tools/llama/build_dataset.py index 7b4273b2..20e22199 100644 --- a/tools/llama/build_dataset.py +++ b/tools/llama/build_dataset.py @@ -12,7 +12,8 @@ from tqdm import tqdm from fish_speech.datasets.protos.text_data_pb2 import Semantics, Sentence, TextData -from fish_speech.datasets.protos.text_data_stream import load_filelist, pack_pb_stream +from fish_speech.datasets.protos.text_data_stream import pack_pb_stream +from fish_speech.utils.file import load_filelist # To avoid CPU overload os.environ["MKL_NUM_THREADS"] = "1"