pseeth hugggof commited on
Commit
5235095
1 Parent(s): 4d9dee0

fix beat prompt context (#8)

Browse files

- fix beat prompt context (c011c543099bf39ca8b3fb98495a509489efc75f)


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

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,