Fix docstring

facebookresearch · Sep 23, 2023 · e42de6f · e42de6f
1 parent a2b9675
commit e42de6f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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`.