fix beat prompt context (#8)

- fix beat prompt context (c011c543099bf39ca8b3fb98495a509489efc75f)


Co-authored-by: Hugo Flores <hugggof@users.noreply.huggingface.co>

vampnet/interface.py

@@ -194,8 +194,8 @@ class Interface(torch.nn.Module):
 
     def make_beat_mask(self, 
             signal: AudioSignal, 
-            before_beat_s: float = 0.1,
-            after_beat_s: float = 0.1,
+            before_beat_s: float = 0.0,
+            after_beat_s: float = 0.02,
             mask_downbeats: bool = True,
             mask_upbeats: bool = True,
             downbeat_downsample_factor: int = None,