deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

man I'm really on it tonight....

Browse Source
This commit is contained in:
James Betker 2022-04-02 22:01:33 -06:00
parent a4bdc80933
commit 99de63a922
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/audio/tts/autoregressive_codegen.py
View File

@ -217,7 +217,7 @@ class AutoregressiveCodegen(nn.Module):
ff_mult=1,
rotary_pos_emb=True,
))
self.encoder.to_logits = nn.Identity() # This is unused.
self.encoder.transformer.to_logits = nn.Identity() # This is unused.
self.decoder = CheckpointedXTransformerWrapper(
num_tokens=num_mel_tokens,
use_pos_emb=False,