From a4bdc8093315b293fb53c14c02552a9d729e9d53 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sat, 2 Apr 2022 21:59:50 -0600 Subject: [PATCH] moikmadsf --- codes/models/audio/tts/autoregressive_codegen.py | 1 + 1 file changed, 1 insertion(+) diff --git a/codes/models/audio/tts/autoregressive_codegen.py b/codes/models/audio/tts/autoregressive_codegen.py index 9f2198af..07429a5e 100644 --- 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,