From 8f3c3e01eee220d1899387406b743f0c679d6d5a Mon Sep 17 00:00:00 2001
From: mrq <mrq@ecker.tech>
Date: Sun, 23 Feb 2025 12:09:56 -0600
Subject: [PATCH] oops

---
 vall_e/models/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vall_e/models/base.py b/vall_e/models/base.py
index 4c8217e..ca1aa54 100755
--- a/vall_e/models/base.py
+++ b/vall_e/models/base.py
@@ -736,7 +736,7 @@ class Base(nn.Module):
 			self.audio_decoder = AudioDecoder(
 				d_model,
 				d_model * 2,
-				(n_audio_tokens + 2) * self.n_resp_levels,
+				(n_audio_tokens + 1) * self.n_resp_levels,
 			)
 
 		if attention_backend == "auto":