floppyrj45/lingbot-map
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
LinZhuoChen
2026-04-16 21:05:39 +08:00
parent 223d71c7ec
commit c1b833aaee
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
demo.py
View File

@@ -254,7 +254,7 @@ def main():
default=1, default=1,
help="Streaming only. Every N-th frame after scale frames is kept as a keyframe. 1 = every frame.", help="Streaming only. Every N-th frame after scale frames is kept as a keyframe. 1 = every frame.",
) )
parser.add_argument("--kv_cache_sliding_window", type=int, default=64) parser.add_argument("--kv_cache_sliding_window", type=int, default=16)
parser.add_argument("--kv_cache_scale_frames", type=int, default=8) parser.add_argument("--kv_cache_scale_frames", type=int, default=8)
parser.add_argument("--use_sdpa", action="store_true", default=False, parser.add_argument("--use_sdpa", action="store_true", default=False,
help="Use SDPA backend (no flashinfer needed). Default: FlashInfer") help="Use SDPA backend (no flashinfer needed). Default: FlashInfer")
@@ -340,8 +340,7 @@ def main():
num_scale_frames=args.num_scale_frames, num_scale_frames=args.num_scale_frames,
) )
t_infer = time.time() - t0 print(f"Inference done in {time.time() - t0:.1f}s")
print(f"Inference done: {t_infer:.1f}s ({num_frames / t_infer:.1f} FPS)")
# ── Post-process ───────────────────────────────────────────────────────── # ── Post-process ─────────────────────────────────────────────────────────
predictions, images_cpu = postprocess(predictions, images) predictions, images_cpu = postprocess(predictions, images)