[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

+6 -2
View File
@@ -220,16 +220,20 @@ def run_and_validate_text_to_video_request(omni: Omni):
_validate_video(_run_offline_t2v(omni))
def run_and_validate_image_to_video_request(omni: Omni):
def run_and_validate_image_to_video_request(omni: Omni, check_t2v_divergence: bool = True):
"""Run and validate an image to video request.
LTX2Pipeline uses a unified text/image entry and can fall back to T2V if
the image never reaches conditioning, so also run T2V with the same seed
and prompt and assert the outputs differ; a regression that silently
drops the input image would otherwise still pass a "did we get a valid
video back" check alone.
video back" check alone. Pipelines that mandate an image and fail closed
without one (SANA-Video I2V) pass check_t2v_divergence=False: a dropped
image raises instead of silently degrading, so the T2V comparison is moot.
"""
i2v_video = _validate_video_frames(_get_offline_videos(_run_offline_i2v(omni)))[0]
if not check_t2v_divergence:
return
t2v_video = _validate_video_frames(_get_offline_videos(_run_offline_t2v(omni)))[0]
assert not np.array_equal(i2v_video, t2v_video), (
"I2V output is identical to T2V output with the same seed and prompt; "