Просмотр исходного кода

Update webui.py (#699)

* Update webui.py

* [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>
spicysama 1 год назад
Родитель
Сommit
30e1762594
1 измененных файлов с 2 добавлено и 1 удалено
  1. 2 1
      tools/webui.py

+ 2 - 1
tools/webui.py

@@ -23,7 +23,7 @@ from fish_speech.i18n import i18n
 from fish_speech.text.chn_text_norm.text import Text as ChnNormedText
 from fish_speech.text.chn_text_norm.text import Text as ChnNormedText
 from fish_speech.utils import autocast_exclude_mps, set_seed
 from fish_speech.utils import autocast_exclude_mps, set_seed
 from tools.api import decode_vq_tokens, encode_reference
 from tools.api import decode_vq_tokens, encode_reference
-from tools.file import AUDIO_EXTENSIONS, list_files
+from tools.file import AUDIO_EXTENSIONS, audio_to_bytes, list_files, read_ref_text
 from tools.llama.generate import (
 from tools.llama.generate import (
     GenerateRequest,
     GenerateRequest,
     GenerateResponse,
     GenerateResponse,
@@ -87,6 +87,7 @@ def build_html_error_message(error):
 def inference(req: ServeTTSRequest):
 def inference(req: ServeTTSRequest):
 
 
     idstr: str | None = req.reference_id
     idstr: str | None = req.reference_id
+    prompt_tokens, prompt_texts = [], []
     if idstr is not None:
     if idstr is not None:
         ref_folder = Path("references") / idstr
         ref_folder = Path("references") / idstr
         ref_folder.mkdir(parents=True, exist_ok=True)
         ref_folder.mkdir(parents=True, exist_ok=True)