Spaces:

descript
/

vampnet

Runtime error

Hugo Flores Garcia commited on Jul 19, 2023

Commit

8c3b3e7

1 Parent(s): 9fdab57

beat prompt ctx

Files changed (1) hide show

vampnet/interface.py CHANGED Viewed

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

     def make_beat_mask(self,
             signal: AudioSignal,
             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,