From 99de63a9222f9122aae0c73ec4fb7be696cca276 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sat, 2 Apr 2022 22:01:33 -0600 Subject: [PATCH] man I'm really on it tonight.... --- codes/models/audio/tts/autoregressive_codegen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/audio/tts/autoregressive_codegen.py b/codes/models/audio/tts/autoregressive_codegen.py index 07429a5e..7831f1de 100644 --- a/codes/models/audio/tts/autoregressive_codegen.py +++ b/codes/models/audio/tts/autoregressive_codegen.py @@ -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,