Re-add start_text_token

This commit is contained in:
James Betker 2021-12-22 14:10:35 -07:00
parent a9629f7022
commit 66bc60aeff

View File

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