Hugo Flores Garcia commited on
Commit
8c3b3e7
1 Parent(s): 9fdab57

beat prompt ctx

Browse files
Files changed (1) hide show
  1. vampnet/interface.py +1 -1
vampnet/interface.py CHANGED
@@ -195,7 +195,7 @@ class Interface(torch.nn.Module):
195
  def make_beat_mask(self,
196
  signal: AudioSignal,
197
  before_beat_s: float = 0.0,
198
- after_beat_s: float = 0.1,
199
  mask_downbeats: bool = True,
200
  mask_upbeats: bool = True,
201
  downbeat_downsample_factor: int = None,
 
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,