- 01 May, 2025 1 commit
-
-
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:muyangli <lmxyy1999@foxmail.com>
-
- 04 Apr, 2025 2 commits
-
-
sxtyzhangzk authored
-
Hyunsung Lee authored
-
- 01 Apr, 2025 3 commits
-
-
sxtyzhangzk authored
-
sxtyzhangzk authored
-
Hyunsung Lee authored
-
- 07 Mar, 2025 1 commit
-
-
muyangli authored
Co-authored-by:
Zhekai Zhang <sxtyzhangzk@gmail.com> Co-authored-by:
Muyang Li <lmxyy1999@foxmail.com> Co-authored-by:
Yujun Lin <16437040+synxlin@users.noreply.github.com>
-
- 20 Feb, 2025 1 commit
-
-
muyangli authored
-
- 11 Nov, 2024 1 commit
-
-
sxtyzhangzk authored
-
- 08 Nov, 2024 1 commit
-
-
Zhekai Zhang authored
-