mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

ugh

Browse Source
This commit is contained in:
mrq 2024-06-08 20:34:00 -05:00
parent 8d068fa3f9
commit 5c732b72ee
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vall_e/models/base.py
View File

@ -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