Spaces:
Sleeping
Sleeping
fix beat prompt context
Browse files- 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.
|
198 |
-
after_beat_s: float = 0.
|
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,
|