update doc

facebookresearch · Nov 5, 2024 · dbbf222 · dbbf222
1 parent 123aa8d
commit dbbf222
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/audiocraft/models/lm.py b/audiocraft/models/lm.py
@@ -350,6 +350,8 @@ def _sample_next_token(self,
                 in paragraph 4.3 (https://arxiv.org/pdf/2407.12563). This beta coefficient is meant to
                 push the text condition more than the style condition in the case where both text and style
                 conditions are being used.
+            two_step_cfg (bool): Whether to run classifier free-guidance with 2 distinct steps.
+
         Returns:
             next_token (torch.Tensor): Next token tensor of shape [B, K, 1].
         """