Spaces:
Runtime error
Runtime error
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,
|