From 2fbeacfe92d22a31de726e8fe66c719b879e8502 Mon Sep 17 00:00:00 2001 From: mrq Date: Thu, 14 Nov 2024 22:18:33 -0600 Subject: [PATCH] ugh --- vall_e/models/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vall_e/models/base.py b/vall_e/models/base.py index 3fa6037..5aef519 100755 --- a/vall_e/models/base.py +++ b/vall_e/models/base.py @@ -471,9 +471,9 @@ class Base(nn.Module): resp_l_names = ['AR:0:0'] + [f'NAR:{i}:{i+1}' for i in range( self.n_resp_levels - 1 )] classifier_l_tokens = l_tokens + [ n_text_tokens ] - classifier_l_names = resp_l_names + [ "len" ] + classifier_l_names = resp_l_names + [ "stt" ] - if "len" in self.capabilities and False: + if "len" in self.capabilities: classifier_l_tokens += [ n_text_tokens ] classifier_l_names += ["len"]