• Tran Thanh Luan's avatar
    [Feat] TaylorSeer Cache (#12648) · 6290fdfd
    Tran Thanh Luan authored
    
    
    * init taylor_seer cache
    
    * make compatible with any tuple size returned
    
    * use logger for printing, add warmup feature
    
    * still update in warmup steps
    
    * refractor, add docs
    
    * add configurable cache, skip compute module
    
    * allow special cache ids only
    
    * add stop_predicts (cooldown)
    
    * update docs
    
    * apply ruff
    
    * update to handle multple calls per timestep
    
    * refractor to use state manager
    
    * fix format & doc
    
    * chores: naming, remove redundancy
    
    * add docs
    
    * quality & style
    
    * fix taylor precision
    
    * Apply style fixes
    
    * add tests
    
    * Apply style fixes
    
    * Remove TaylorSeerCacheTesterMixin from flux2 tests
    
    * rename identifiers, use more expressive taylor predict loop
    
    * torch compile compatible
    
    * Apply style fixes
    
    * Update src/diffusers/hooks/taylorseer_cache.py
    Co-authored-by: default avatarDhruv Nair <dhruv.nair@gmail.com>
    
    * update docs
    
    * make fix-copies
    
    * fix example usage.
    
    * remove tests on flux kontext
    
    ---------
    Co-authored-by: default avatartoilaluan <toilaluan@github.com>
    Co-authored-by: default avatargithub-actions[bot] <github-actions[bot]@users.noreply.github.com>
    Co-authored-by: default avatarDhruv Nair <dhruv.nair@gmail.com>
    Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
    6290fdfd
cache.md 1.23 KB