[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:
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,
|
||||
),
|
||||
|
||||
Reference in new issue
Block a user