adityas129 commited on
Commit
d413390
·
verified ·
1 Parent(s): 0aab971

Update jam_worker.py

Browse files
Files changed (1) hide show
  1. 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