storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-25 08:40:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Made some fixes to the container.

Browse Source
This commit is contained in:
Michael Chung
2023-10-23 19:58:25 -04:00
parent e685ace23a
commit 45a50d2dba
2 changed files with 28 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/VALL-E-X/utils/prompt_making.py
+1
View File
@@ -84,6 +84,7 @@ def make_prompt(name:pathlib.Path, audio_prompt_path:pathlib.Path,audio_prompt_o
else:
save_path = os.path.join("./customs/", f"{name}.npz")
np.savez(save_path, audio_tokens=audio_tokens, text_tokens=text_tokens, lang_code=lang2code[lang_pr])
print(f"Embedding Save Path {save_path}")
logging.info(f"Successful. Prompt saved to {save_path}")
def make_transcript(name, wav, sr, whisper_folder_path: pathlib.Path,transcript=None):