mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

moikmadsf

Browse Source
This commit is contained in:
James Betker 2022-04-02 21:59:50 -06:00
parent 1cf20b7337
commit a4bdc80933
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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