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

Rename variables and add comments

Browse Source
This commit is contained in:
Zygimantas Straznickas
2023-04-22 12:23:55 -07:00
parent 15606ed12f
commit bee9e03080
3 changed files with 24 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bark/generation.py
View File

@@ -428,7 +428,7 @@ def generate_text_semantic(
else:
x_input = x
logits, kv_cache = model(x_input, merge_context=True, use_cache=use_kv_caching, past_key_values=kv_cache)
logits, kv_cache = model(x_input, merge_context=True, use_cache=use_kv_caching, past_kv=kv_cache)
relevant_logits = logits[0, 0, :SEMANTIC_VOCAB_SIZE]
if allow_early_stop:
relevant_logits = torch.hstack(
@@ -611,7 +611,7 @@ def generate_coarse(
else:
x_input = x_in
logits, kv_cache = model(x_input, use_cache=use_kv_caching, past_key_values=kv_cache)
logits, kv_cache = model(x_input, use_cache=use_kv_caching, past_kv=kv_cache)
logit_start_idx = (
SEMANTIC_VOCAB_SIZE + (1 - int(is_major_step)) * CODEBOOK_SIZE
)