Skip to content

Commit 87d4978

Browse files
committed
Update haven_vlm_connector.py
1 parent 0d5c272 commit 87d4978

1 file changed

Lines changed: 13 additions & 4 deletions

File tree

plugins/AHavenVLMConnector/haven_vlm_connector.py

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535
"stashapi:stashapp-tools==0.2.58",
3636
"aiohttp==3.12.13",
3737
"pydantic==2.12.5",
38-
"vlm-engine==0.9.6",
38+
"vlm-engine==1.0.0",
3939
"pyyaml==6.0.2",
4040
)
4141

@@ -355,11 +355,20 @@ def progress_cb(p: int) -> None:
355355
total_prog = sum(video_progress.values()) / total_tasks
356356

357357
stats = vlm_engine.vlm_engine.get_performance_stats()
358-
last_rt = stats.get("last_response_time", 0.0)
359-
fps = 1.0 / last_rt if last_rt > 0 else 0.0
358+
total_frames = stats.get("total_frames_processed", 0)
359+
elapsed_seconds = stats.get("elapsed_time", 0.0)
360360

361+
log.info(f"[Throughput] total_frames: {total_frames}")
362+
log.info(f"[Throughput] elapsed_seconds: {elapsed_seconds:.2f}")
363+
364+
if elapsed_seconds > 0:
365+
fpm = (total_frames / elapsed_seconds) * 60.0
366+
else:
367+
fpm = 0.0
368+
369+
log.info(f"[Throughput] calculated_fpm: {fpm:.1f}")
361370
log.info(
362-
f"[FPS] Frame ~{(p / 100) * 100:.0f}: {fps:.2f} fps | progress: {p}%"
371+
f"[Throughput] Frame ~{(p / 100) * 100:.0f}: {fpm:.1f} FPM | progress: {p}%"
363372
)
364373
log.progress(total_prog)
365374

0 commit comments

Comments
 (0)