Skip to content

Add progress feedback during model warmup in streaming demo - #65

Open
1TommyCheung wants to merge 1 commit into
Soul-AILab:mainfrom
1TommyCheung:fix/streaming-warmup-progress
Open

Add progress feedback during model warmup in streaming demo#65
1TommyCheung wants to merge 1 commit into
Soul-AILab:mainfrom
1TommyCheung:fix/streaming-warmup-progress

Conversation

@1TommyCheung

Copy link
Copy Markdown

Summary

  • Adds descriptive progress messages during cold start phases in gradio_app_streaming.py
  • On first run, torch.compile takes ~1-2 minutes with no UI feedback, making the app appear frozen
  • Progress bar now shows what's happening at each warmup stage: model loading, VAE compilation, CUDA kernel compilation
  • Subsequent runs (compilation cached) show the original short progress messages
  • No changes to generation behavior or streaming output

Test plan

  • Launch gradio_app_streaming.py with a fresh model (no torch compile cache)
  • Verify progress messages appear during warmup phases
  • Verify progress disappears once video streaming begins
  • Run a second generation and confirm shorter progress messages are shown

🤖 Generated with Claude Code

The first run of the streaming Gradio app appears frozen for ~1-2 minutes
while torch.compile compiles CUDA kernels. This adds descriptive progress
messages during cold start phases (model loading, VAE compilation, kernel
compilation) so users know the app is working. Subsequent runs show the
original short messages since compilation is cached.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant