From c9b9d706390e9c3f22f6e24a40222629e788e337 Mon Sep 17 00:00:00 2001 From: Antoine Drabble Date: Thu, 5 Oct 2023 10:26:12 +0200 Subject: [PATCH] Fix docstring Match the acronym in `codebooks_patterns.py`. --- audiocraft/modules/codebooks_patterns.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/audiocraft/modules/codebooks_patterns.py b/audiocraft/modules/codebooks_patterns.py index 61362588..43c177d4 100644 --- a/audiocraft/modules/codebooks_patterns.py +++ b/audiocraft/modules/codebooks_patterns.py @@ -30,7 +30,7 @@ class Pattern: The pattern provides convenient methods to build and revert interleaved sequences from it: ``build_pattern_sequence`` maps a given a dense input tensor of multi-codebook sequence from [B, K, T] - to the interleaved sequence of shape [B, K, S] applying the pattern, with S being the batch size, + to the interleaved sequence of shape [B, K, S] applying the pattern, with B being the batch size, K being the number of codebooks, T the number of original timesteps and S the number of sequence steps for the output sequence. The unfilled positions are replaced with a special token and the built sequence is returned along with a mask indicating valid tokens.