Commit
·
7b1b186
1
Parent(s):
5dd53c8
attempting a prime step for more reliable jam starts
Browse files- jam_worker.py +16 -1
jam_worker.py
CHANGED
|
@@ -36,6 +36,7 @@ class JamParams:
|
|
| 36 |
temperature: float = 1.1
|
| 37 |
topk: int = 40
|
| 38 |
style_ramp_seconds: float = 8.0 # 0 => instant (current behavior), try 6.0–10.0 for gentle glides
|
|
|
|
| 39 |
|
| 40 |
|
| 41 |
@dataclass
|
|
@@ -533,7 +534,21 @@ class JamWorker(threading.Thread):
|
|
| 533 |
# ---------- main loop ----------
|
| 534 |
|
| 535 |
def run(self):
|
| 536 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 537 |
while not self._stop_event.is_set():
|
| 538 |
# throttle generation if we are far ahead
|
| 539 |
if not self._should_generate_next_chunk():
|
|
|
|
| 36 |
temperature: float = 1.1
|
| 37 |
topk: int = 40
|
| 38 |
style_ramp_seconds: float = 8.0 # 0 => instant (current behavior), try 6.0–10.0 for gentle glides
|
| 39 |
+
prime_steps: int = 1
|
| 40 |
|
| 41 |
|
| 42 |
@dataclass
|
|
|
|
| 534 |
# ---------- main loop ----------
|
| 535 |
|
| 536 |
def run(self):
|
| 537 |
+
# --- PRIME (optional) ---
|
| 538 |
+
n_prime = max(0, int(getattr(self.params, "prime_steps", 0) or 0))
|
| 539 |
+
if n_prime:
|
| 540 |
+
with self._lock:
|
| 541 |
+
target = self.params.style_vec
|
| 542 |
+
if target is not None:
|
| 543 |
+
# initialize or nudge _style_vec exactly once before prime
|
| 544 |
+
self._style_vec = np.array(target, dtype=np.float32, copy=True)
|
| 545 |
+
style_to_use = self._style_vec
|
| 546 |
+
|
| 547 |
+
for _ in range(n_prime):
|
| 548 |
+
_wav, self.state = self.mrt.generate_chunk(state=self.state, style=style_to_use)
|
| 549 |
+
# discard audio; just advance state
|
| 550 |
+
|
| 551 |
+
|
| 552 |
while not self._stop_event.is_set():
|
| 553 |
# throttle generation if we are far ahead
|
| 554 |
if not self._should_generate_next_chunk():
|