From b6afc4d542b5c25f291d12837e691bce741b9411 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sat, 2 Apr 2022 21:57:00 -0600 Subject: [PATCH] dsfa --- codes/models/audio/tts/autoregressive_codegen.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/codes/models/audio/tts/autoregressive_codegen.py b/codes/models/audio/tts/autoregressive_codegen.py index b4003aae..b11764ea 100644 --- a/codes/models/audio/tts/autoregressive_codegen.py +++ b/codes/models/audio/tts/autoregressive_codegen.py @@ -205,6 +205,7 @@ class AutoregressiveCodegen(nn.Module): self.encoder = CheckpointedXTransformerWrapper( num_tokens=num_text_tokens, use_pos_emb=False, + max_seq_len=-1, attn_layers = Encoder( depth=depth//2, heads=model_dim//64, @@ -220,6 +221,7 @@ class AutoregressiveCodegen(nn.Module): self.decoder = CheckpointedXTransformerWrapper( num_tokens=num_mel_tokens, use_pos_emb=False, + max_seq_len=-1, attn_layers=Decoder( depth=depth, heads=model_dim//64,