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

disabled chunkwise_recurrent for 2x speed gains (I suppose it has been working the entire time, but I have not been properly grabbing things, and this might explain why the output is bad)

Browse Source
This commit is contained in:
mrq 2023-08-25 19:50:19 -05:00
parent 6455a2f9d7
commit 16e0020901
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

@ -171,7 +171,7 @@ class Base(nn.Module):
dropout=p_dropout, dropout=p_dropout,
checkpoint_activations=True, checkpoint_activations=True,
chunkwise_recurrent=self.causal, chunkwise_recurrent=False, # self.causal,
recurrent_chunkwise_size=64, recurrent_chunkwise_size=64,
no_output_layer=True, no_output_layer=True,
decoder_normalize_before=True, decoder_normalize_before=True,