Skip to content

Conversation

@ryanontheinside
Copy link
Collaborator

feat: add VAE download infrastructure with metadata registry

Adds complete infrastructure for downloading and managing VAE models independently of pipeline downloads. This change has no effect on the current code. When the ability to select different VAE types is added, the system will check if the selected VAE is downloaded and prompt for download if missing.

Backend:

  • Add /api/v1/vae/status endpoint to check VAE download status
  • Add /api/v1/vae/download endpoint to trigger VAE downloads
  • Add VAEMetadata dataclass and VAE_METADATA registry as single source of truth for VAE filenames and download sources
  • Add vae_file_exists() and get_vae_file_path() using metadata registry
  • Add download_vae() and download_downloadable_vaes() functions

Frontend:

  • Add checkVaeStatus() and downloadVae() API functions
  • Extend DownloadDialog to show VAE-specific download prompts
  • Add VAE status checking before stream start in StreamPage
  • Add separate VAE download flow with polling for completion

Prepares codebase for upcoming PRs adding additional VAE types.

feat: add VAE download infrastructure with metadata registry

Adds complete infrastructure for downloading and managing VAE models
independently of pipeline downloads. This change has no effect on the current code. When the ability to select different VAE types is added, the system will check if the selected VAE is downloaded and
prompt for download if missing.

Backend:
- Add /api/v1/vae/status endpoint to check VAE download status
- Add /api/v1/vae/download endpoint to trigger VAE downloads
- Add VAEMetadata dataclass and VAE_METADATA registry as single source
  of truth for VAE filenames and download sources
- Add vae_file_exists() and get_vae_file_path() using metadata registry
- Add download_vae() and download_downloadable_vaes() functions

Frontend:
- Add checkVaeStatus() and downloadVae() API functions
- Extend DownloadDialog to show VAE-specific download prompts
- Add VAE status checking before stream start in StreamPage
- Add separate VAE download flow with polling for completion

Prepares codebase for upcoming PRs adding additional VAE types.

Signed-off-by: RyanOnTheInside <7623207+ryanontheinside@users.noreply.github.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.

2 participants