Skip to content

Conversation

@RakshithaKowlikar
Copy link

What does this PR do?

This PR adds optional Flashpack integration for significantly faster model weight loading.
Read this blog to understand what flashpack is.

  • adds use_flashpack parameter to enable flashpack loading
  • automatic graceful fallback to safetensors if flashpack is unavailable or incompatible
  • supports variant parameter for specific weight files (e.g., model.fp16.flashpack)
  • feature is optional via use_flashpack=True

Fixes # (12550)

Requires optional dependency: pip install flashpack

flashpack repository: https://github.com/fal-ai/flashpack

@sayakpaul sayakpaul requested a review from DN6 November 24, 2025 12:43
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