[Model] Add Cache-DiT and CPU offload support for SANA-Video 2B (#5882)

Signed-off-by: cr-gao <gaochenrui@sjtu.edu.cn>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
Co-authored-by: Alicia <115451386+congw729@users.noreply.github.com>
This commit is contained in:
authored and GitHub committed 2026-09-02 23:40:48 +08:00
1 parent 95a9311696
commit 3b6ce98421
10 files changed
+558 -26

No files matched your search

@@ -57,6 +57,11 @@ DIFFUSION_TEST_SETTINGS = {
[DiffusionAccs.TENSOR_PARALLEL],
[DiffusionAccs.CFG_PARALLEL],
[DiffusionAccs.TENSOR_PARALLEL, DiffusionAccs.CFG_PARALLEL],
[DiffusionAccs.CACHE_DIT],
[DiffusionAccs.CPU_OFFLOAD],
[DiffusionAccs.LAYERWISE_OFFLOAD],
[DiffusionAccs.CACHE_DIT, DiffusionAccs.CPU_OFFLOAD],
[DiffusionAccs.CACHE_DIT, DiffusionAccs.LAYERWISE_OFFLOAD],
],
check_multi_output=False,
),