From ed174c589e9022a7e12e2152c8d5c6c5ec68bcb9 Mon Sep 17 00:00:00 2001 From: mrq Date: Thu, 7 Nov 2024 09:19:21 -0600 Subject: [PATCH] ugh --- 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 954907b..32bbb16 100755 --- a/vall_e/models/base.py +++ b/vall_e/models/base.py @@ -485,7 +485,7 @@ class Base(nn.Module): # it would be nicer for these to be a token or live inside an embedding self.sep = nn.Parameter(torch.randn(d_model)) self.dropout_token = nn.Parameter(torch.randn(d_model)) - self.mask_token = dropout_token # alias (hopefully) to the above + self.mask_token = self.dropout_token # alias (hopefully) to the above if self.version == 1: # legacy n_audio_tokens += (n_tasks - 1) # old models have the task tokens in the prom