mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Re-add start_text_token

Browse Source
This commit is contained in:
James Betker 2021-12-22 14:10:35 -07:00
parent a9629f7022
commit 66bc60aeff
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/models/gpt_voice/gpt_tts_hf.py
View File

@ -40,6 +40,7 @@ class ConditioningEncoder(nn.Module):
class GptTtsHf(nn.Module): class GptTtsHf(nn.Module):
NUMBER_TEXT_TOKENS = 50257 # The number of BPE tokens produced by the HF GPT2Tokenizer NUMBER_TEXT_TOKENS = 50257 # The number of BPE tokens produced by the HF GPT2Tokenizer
START_TEXT_TOKEN = 50256
STOP_TEXT_TOKEN = 0 STOP_TEXT_TOKEN = 0
NUMBER_MEL_CODES = 8194 NUMBER_MEL_CODES = 8194
START_MEL_TOKEN = 8192 START_MEL_TOKEN = 8192