1. 01 May, 2025 1 commit
    • Andrea Ferretti's avatar
      feat: expose norm1 layer to support TeaCache (#234) · b4d3f50b
      Andrea Ferretti authored
      
      
      * feat: expose norm1 layer to support TeaCache
      
      * feat: add TeaCache example
      
      * feat: add idx as optional parameter
      
      * chore: rename function
      
      * refactor: move TeaCache decorator into example script
      
      * test: add a test for the combination of Nunchaku with TeaCache
      
      * feat: expose norm1 layer to support TeaCache
      
      * feat: add TeaCache example
      
      * feat: add idx as optional parameter
      
      * chore: rename function
      
      * refactor: move TeaCache decorator into example script
      
      * test: add a test for the combination of Nunchaku with TeaCache
      
      * fix: make tests run on low memory hardware
      
      * fix: ensure that memory is correctly released between tests
      
      * fix: avoid moving pipeline to device prematurely
      
      * gpu memory does not release
      
      * need to figure out a way to get compatible with offloading
      
      * wrap up the teacache
      
      ---------
      Co-authored-by: default avatarmuyangli <lmxyy1999@foxmail.com>
      b4d3f50b
  2. 29 Apr, 2025 7 commits
  3. 27 Apr, 2025 8 commits
  4. 26 Apr, 2025 6 commits
  5. 25 Apr, 2025 7 commits
  6. 24 Apr, 2025 5 commits
  7. 23 Apr, 2025 6 commits