From 66bc60aeffa8dde00c705856180203b8d5b4c6f6 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Wed, 22 Dec 2021 14:10:35 -0700
Subject: [PATCH] Re-add start_text_token

---
 codes/models/gpt_voice/gpt_tts_hf.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/codes/models/gpt_voice/gpt_tts_hf.py b/codes/models/gpt_voice/gpt_tts_hf.py
index 2d1b6b06..b804d0d8 100644
--- a/codes/models/gpt_voice/gpt_tts_hf.py
+++ b/codes/models/gpt_voice/gpt_tts_hf.py
@@ -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