fix beat prompt context

#8
by hugggof - opened
Files changed (1) hide show
  1. vampnet/interface.py +2 -2
vampnet/interface.py CHANGED
@@ -194,8 +194,8 @@ class Interface(torch.nn.Module):
194
 
195
  def make_beat_mask(self,
196
  signal: AudioSignal,
197
- before_beat_s: float = 0.1,
198
- after_beat_s: float = 0.1,
199
  mask_downbeats: bool = True,
200
  mask_upbeats: bool = True,
201
  downbeat_downsample_factor: int = None,
 
194
 
195
  def make_beat_mask(self,
196
  signal: AudioSignal,
197
+ before_beat_s: float = 0.0,
198
+ after_beat_s: float = 0.02,
199
  mask_downbeats: bool = True,
200
  mask_upbeats: bool = True,
201
  downbeat_downsample_factor: int = None,