From 5c732b72eec2fa653e452b47f0c440056579955f Mon Sep 17 00:00:00 2001 From: mrq Date: Sat, 8 Jun 2024 20:34:00 -0500 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 ffbb7e9..9e4e4e3 100755 --- a/vall_e/models/base.py +++ b/vall_e/models/base.py @@ -231,7 +231,7 @@ class Base(nn.Module): @property def stop_token(self): - if not self.causal: + if not self.causal or "len" not in self.capabilities: raise ValueError("Not using stop token!") return self.n_audio_tokens