MelodyFlowtry2production

Paused

Gael Le Lan commited on Dec 9, 2024

Commit

f8df20b

1 Parent(s): 9d0d223

fix default CFG

Files changed (1) hide show

audiocraft/models/flow.py CHANGED Viewed

@@ -165,7 +165,7 @@ class FlowModel(torch.nn.Module):
         weight_init: tp.Optional[str] = None,
         depthwise_init: tp.Optional[str] = None,
         zero_bias_init: bool = False,
-        cfg_coef: float = 1.0,
         **kwargs
     ):
         super().__init__()

         weight_init: tp.Optional[str] = None,
         depthwise_init: tp.Optional[str] = None,
         zero_bias_init: bool = False,
+        cfg_coef: float = 4.0,
         **kwargs
     ):
         super().__init__()