Update jam_worker.py
Browse files- jam_worker.py +14 -0
jam_worker.py
CHANGED
|
@@ -156,6 +156,12 @@ class JamWorker(threading.Thread):
|
|
| 156 |
self._stop_event = threading.Event()
|
| 157 |
self._max_buffer_ahead = 1
|
| 158 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 159 |
# reseed queues (install at next bar boundary after emission)
|
| 160 |
self._pending_reseed: Optional[dict] = None # legacy full reset path (kept for fallback)
|
| 161 |
self._pending_token_splice: Optional[dict] = None # seamless token splice
|
|
@@ -738,7 +744,15 @@ class JamWorker(threading.Thread):
|
|
| 738 |
self._style_vec += step * (target.astype(np.float32, copy=False) - self._style_vec)
|
| 739 |
style_to_use = self._style_vec
|
| 740 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 741 |
wav, self.state = self.mrt.generate_chunk(state=self.state, style=style_to_use)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 742 |
# append and spool
|
| 743 |
self._append_model_chunk_and_spool(wav)
|
| 744 |
# try emitting zero or more chunks if available
|
|
|
|
| 156 |
self._stop_event = threading.Event()
|
| 157 |
self._max_buffer_ahead = 1
|
| 158 |
|
| 159 |
+
# activity tracking for session monitoring
|
| 160 |
+
self.created_at = time.time()
|
| 161 |
+
self.last_chunk_started_at: Optional[float] = None
|
| 162 |
+
self.last_chunk_completed_at: Optional[float] = None
|
| 163 |
+
self.last_activity_at = time.time()
|
| 164 |
+
|
| 165 |
# reseed queues (install at next bar boundary after emission)
|
| 166 |
self._pending_reseed: Optional[dict] = None # legacy full reset path (kept for fallback)
|
| 167 |
self._pending_token_splice: Optional[dict] = None # seamless token splice
|
|
|
|
| 744 |
self._style_vec += step * (target.astype(np.float32, copy=False) - self._style_vec)
|
| 745 |
style_to_use = self._style_vec
|
| 746 |
|
| 747 |
+
# Track chunk generation timing
|
| 748 |
+
self.last_chunk_started_at = time.time()
|
| 749 |
+
self.last_activity_at = time.time()
|
| 750 |
+
|
| 751 |
wav, self.state = self.mrt.generate_chunk(state=self.state, style=style_to_use)
|
| 752 |
+
|
| 753 |
+
self.last_chunk_completed_at = time.time()
|
| 754 |
+
self.last_activity_at = time.time()
|
| 755 |
+
|
| 756 |
# append and spool
|
| 757 |
self._append_model_chunk_and_spool(wav)
|
| 758 |
# try emitting zero or more chunks if available
|