[Model] Add TP and CFG parallelism to SANA-Video 2B (#5861)

Signed-off-by: cr-gao <gaochenrui@sjtu.edu.cn>
This commit is contained in:
vOv authored and GitHub committed 2026-09-02 17:44:41 +08:00
1 parent b6e72f3879
commit 57c08d34c3
18 files changed
+1621 -116

No files matched your search

@@ -143,7 +143,16 @@ class TestPipelineArgumentsHandling:
@pytest.mark.parametrize(
"feature_id",
["cfg_parallel", "ulysses", "ring", "teacache", "cache_dit", "enforce_eager", "unsupported_quantization"],
[
"cfg_parallel",
"tensor_parallel",
"ulysses",
"ring",
"teacache",
"cache_dit",
"enforce_eager",
"unsupported_quantization",
],
)
def test_adapter_guard_unsupported_feature(self, feature_id):
if feature_id == "cfg_parallel":
@@ -151,6 +160,11 @@ class TestPipelineArgumentsHandling:
parallel_config=DiffusionParallelConfig(cfg_parallel_size=2, sequence_parallel_size=1),
cache_backend="none",
)
elif feature_id == "tensor_parallel":
od_config = _make_od_config(
parallel_config=DiffusionParallelConfig(tensor_parallel_size=2, sequence_parallel_size=1),
cache_backend="none",
)
elif feature_id == "ulysses":
od_config = _make_od_config(
parallel_config=DiffusionParallelConfig(cfg_parallel_size=1, ulysses_degree=2),