moikmadsf

2022-04-02 21:59:50 -06:00 · 2022-04-02 21:59:50 -06:00 · a4bdc80933
commit a4bdc80933
parent 1cf20b7337
1 changed files with 1 additions and 0 deletions
--- a/codes/models/audio/tts/autoregressive_codegen.py
+++ b/codes/models/audio/tts/autoregressive_codegen.py
@ -217,6 +217,7 @@ class AutoregressiveCodegen(nn.Module):
                                      ff_mult=1,
                                      rotary_pos_emb=True,
                                  ))
+        self.encoder.to_logits = nn.Identity()  # This is unused.
        self.decoder = CheckpointedXTransformerWrapper(
                                  num_tokens=num_mel_tokens,
                                  use_pos_emb=False,