ugh
This commit is contained in:
parent
5c732b72ee
commit
80f9530840
|
@ -231,7 +231,7 @@ class Base(nn.Module):
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def stop_token(self):
|
def stop_token(self):
|
||||||
if not self.causal or "len" not in self.capabilities:
|
if not self.causal and "len" not in self.capabilities:
|
||||||
raise ValueError("Not using stop token!")
|
raise ValueError("Not using stop token!")
|
||||||
return self.n_audio_tokens
|
return self.n_audio_tokens
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user