Ruff autofix PLW3301

2023-09-27 00:51:17 +03:00 · 2023-09-27 00:51:17 +03:00 · bc2cf296a3
parent 64bb41f4fa
commit bc2cf296a3
1 changed files with 1 additions and 1 deletions
--- a/TTS/tts/layers/delightful_tts/acoustic_model.py
+++ b/TTS/tts/layers/delightful_tts/acoustic_model.py
@ -362,7 +362,7 @@ class AcousticModel(torch.nn.Module):

        pos_encoding = positional_encoding(
            self.emb_dim,
-            max(token_embeddings.shape[1], max(mel_lens)),
+            max(token_embeddings.shape[1], *mel_lens),
            device=token_embeddings.device,
        )
        encoder_outputs = self.encoder(