n0p/mlx-video
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update video generation completion message to display elapsed time in a more user-friendly format, showing minutes and seconds instead of just seconds.

Browse Source
This commit is contained in:
Prince Canuma
2026-01-19 02:23:51 +01:00
parent 4cd58f8b26
commit e0ee934b99
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mlx_video/generate.py
View File

@@ -1295,8 +1295,10 @@ def generate_video(
console.print(f"[green]✅ Saved {len(video_np)} frames to {frames_dir}[/]") console.print(f"[green]✅ Saved {len(video_np)} frames to {frames_dir}[/]")
elapsed = time.time() - start_time elapsed = time.time() - start_time
minutes, seconds = divmod(elapsed, 60)
time_str = f"{int(minutes)}m {seconds:.1f}s" if minutes >= 1 else f"{seconds:.1f}s"
console.print(Panel( console.print(Panel(
f"[bold green]🎉 Done![/] Generated in {elapsed:.1f}s ({elapsed/num_frames:.2f}s/frame)\n" f"[bold green]🎉 Done![/] Generated in {time_str} ({elapsed/num_frames:.2f}s/frame)\n"
f"[bold green]✨ Peak memory:[/] {mx.get_peak_memory() / (1024 ** 3):.2f}GB", f"[bold green]✨ Peak memory:[/] {mx.get_peak_memory() / (1024 ** 3):.2f}GB",
expand=False expand=False
)) ))