From e42de6fbfdbf149598cd8f69c2ffb2bc91a9cdcb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Carl=20Thom=C3=A9?= <carlthome@gmail.com>
Date: Sat, 23 Sep 2023 18:00:18 +0200
Subject: [PATCH] Fix docstring

---
 audiocraft/models/lm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/audiocraft/models/lm.py b/audiocraft/models/lm.py
index 8cefd2c5..75790e7d 100644
--- a/audiocraft/models/lm.py
+++ b/audiocraft/models/lm.py
@@ -225,7 +225,7 @@ def forward(self, sequence: torch.Tensor,
         S the sequence steps, return the logits with shape [B, card, K, S].
 
         Args:
-            indices (torch.Tensor): Indices of the codes to model.
+            sequence (torch.Tensor): Indices of the codes to model.
             conditions (list of ConditioningAttributes): Conditions to use when modeling
                 the given codes. Note that when evaluating multiple time with the same conditioning
                 you should pre-compute those and pass them as `condition_tensors`.