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. 25 Apr, 2025 1 commit
  3. 23 Apr, 2025 1 commit
  4. 19 Apr, 2025 1 commit
  5. 08 Apr, 2025 1 commit
  6. 04 Apr, 2025 3 commits
  7. 01 Apr, 2025 12 commits
  8. 08 Mar, 2025 2 commits
  9. 07 Mar, 2025 1 commit
  10. 05 Mar, 2025 1 commit
  11. 27 Feb, 2025 1 commit
  12. 22 Feb, 2025 2 commits
  13. 20 Feb, 2025 1 commit
  14. 13 Feb, 2025 1 commit
    • Samuel Tesfai's avatar
      Moving gemm kernels to awq · 91c7b53c
      Samuel Tesfai authored
      Removing torch dependencies in gemm kernel
      Merged tinychat_pybind and pybind
      Merged the load and setup script
      91c7b53c
  15. 11 Feb, 2025 1 commit
  16. 23 Jan, 2025 1 commit
  17. 11 Nov, 2024 6 commits
  18. 10 Nov, 2024 1 commit
  19. 08 Nov, 2024 1 commit