da03 commited on
Commit
9ced953
·
1 Parent(s): 6c00a2b
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -136,6 +136,7 @@ def process_frame(
136
  # Decoding
137
  start = time.perf_counter()
138
  sample = sample_latent * DATA_NORMALIZATION['std'] + DATA_NORMALIZATION['mean']
 
139
  sample = model.decode_first_stage(sample)
140
  sample = sample.squeeze(0).clamp(-1, 1)
141
  timing['decode'] = time.perf_counter() - start
@@ -220,7 +221,7 @@ async def websocket_endpoint(websocket: WebSocket):
220
 
221
  # Print global FPS measurement
222
  print(f" Global FPS: {global_fps:.2f} (total: {frame_count} frames in {total_elapsed:.2f}s)")
223
- time.sleep(10)
224
 
225
  img = Image.fromarray(sample_img)
226
  buffered = io.BytesIO()
 
136
  # Decoding
137
  start = time.perf_counter()
138
  sample = sample_latent * DATA_NORMALIZATION['std'] + DATA_NORMALIZATION['mean']
139
+ time.sleep(10)
140
  sample = model.decode_first_stage(sample)
141
  sample = sample.squeeze(0).clamp(-1, 1)
142
  timing['decode'] = time.perf_counter() - start
 
221
 
222
  # Print global FPS measurement
223
  print(f" Global FPS: {global_fps:.2f} (total: {frame_count} frames in {total_elapsed:.2f}s)")
224
+
225
 
226
  img = Image.fromarray(sample_img)
227
  buffered = io.BytesIO()