Spaces:

facebook
/

MelodyFlow

Running on Zero

Gael Le Lan commited on 12 days ago

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__()