thecollabagepatch commited on
Commit
7b1b186
·
1 Parent(s): 5dd53c8

attempting a prime step for more reliable jam starts

Browse files
Files changed (1) hide show
  1. 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
- # generate until stopped
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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():