Mirrors/bark-with-voice-clone
1
0
Fork 0
mirror of https://github.com/serp-ai/bark-with-voice-clone.git synced 2025-12-16 03:38:01 +01:00
Code Issues Packages Projects Releases Wiki Activity

add more prompts

Browse Source
This commit is contained in:
Georg
2023-04-11 10:48:30 -05:00
parent 20990e257e
commit 7c88255966
18 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
bark/generation.py
View File

@@ -49,10 +49,9 @@ COARSE_RATE_HZ = 75
SAMPLE_RATE = 24_000
ALLOWED_PROMPTS = (
"brylcream",
"es-woman",
"man-narrator",
ALLOWED_PROMPTS = set(
[f"speech_{n}" for n in range(8)] +
[f"music_{n}" for n in range(6)]
)
@@ -307,7 +306,7 @@ def generate_text_semantic(
assert (history_prompt in ALLOWED_PROMPTS)
semantic_history = np.load(
os.path.join(CUR_PATH, "assets", "prompts", f"{history_prompt}.npz")
)["text"]
)["semantic_prompt"]
assert (
isinstance(semantic_history, np.ndarray)
and len(semantic_history.shape) == 1
@@ -452,8 +451,8 @@ def generate_coarse(
x_history = np.load(
os.path.join(CUR_PATH, "assets", "prompts", f"{history_prompt}.npz")
)
x_semantic_history = x_history["coarse_1"]
x_coarse_history = x_history["coarse_2"]
x_semantic_history = x_history["semantic_prompt"]
x_coarse_history = x_history["coarse_prompt"]
assert (
isinstance(x_semantic_history, np.ndarray)
and len(x_semantic_history.shape) == 1
@@ -594,7 +593,7 @@ def generate_fine(
assert (history_prompt in ALLOWED_PROMPTS)
x_fine_history = np.load(
os.path.join(CUR_PATH, "assets", "prompts", f"{history_prompt}.npz")
)["fine"]
)["fine_prompt"]
assert (
isinstance(x_fine_history, np.ndarray)
and len(x_fine_history.shape) == 2