• Aryan's avatar
    First Block Cache (#11180) · 0454fbb3
    Aryan authored
    
    
    * update
    
    * modify flux single blocks to make compatible with cache techniques (without too much model-specific intrusion code)
    
    * remove debug logs
    
    * update
    
    * cache context for different batches of data
    
    * fix hs residual bug for single return outputs; support ltx
    
    * fix controlnet flux
    
    * support flux, ltx i2v, ltx condition
    
    * update
    
    * update
    
    * Update docs/source/en/api/cache.md
    
    * Update src/diffusers/hooks/hooks.py
    Co-authored-by: default avatarDhruv Nair <dhruv.nair@gmail.com>
    
    * address review comments pt. 1
    
    * address review comments pt. 2
    
    * cache context refacotr; address review pt. 3
    
    * address review comments
    
    * metadata registration with decorators instead of centralized
    
    * support cogvideox
    
    * support mochi
    
    * fix
    
    * remove unused function
    
    * remove central registry based on review
    
    * update
    
    ---------
    Co-authored-by: default avatarDhruv Nair <dhruv.nair@gmail.com>
    0454fbb3
hooks.py 10.9 KB