fix typo

facebookresearch · Nov 5, 2024 · 123aa8d · 123aa8d
1 parent 9976586
commit 123aa8d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/audiocraft/models/lm_magnet.py b/audiocraft/models/lm_magnet.py
@@ -125,8 +125,8 @@ def generate(self,
                  top_k: int = 250,
                  top_p: float = 0.0,
                  cfg_coef: tp.Optional[float] = None,
-                 two_step_cfg: tp.Optional[bool] = None,
                  cfg_coef_beta: tp.Optional[float] = None,
+                 two_step_cfg: tp.Optional[bool] = None,
                  remove_prompts: bool = False,
                  check: bool = False,
                  callback: tp.Optional[tp.Callable[[int, int], None]] = None,