[Bugfix][Diffusion] Register Sana I2V pipeline as video output (#6953)

Signed-off-by: wangyu <410167048@qq.com>
This commit is contained in:
wangyu authored and GitHub committed 2026-09-02 23:16:58 +08:00
1 parent 5d20f6b900
commit d5fd9eb6bb
2 files changed
+12

No files matched your search

@@ -64,3 +64,14 @@ def test_video_pipeline_rejects_non_video_final_outputs(stage_configs):
)
def test_registered_video_aliases_declare_video_output(model_class_name):
assert get_diffusion_output_type(model_class_name) == "video"
@pytest.mark.parametrize(
"model_class_name",
[
"SanaVideoPipeline",
"SanaImageToVideoPipeline",
],
)
def test_sana_video_pipelines_declare_video_output(model_class_name):
assert get_diffusion_output_type(model_class_name) == "video"
+1
View File
@@ -88,6 +88,7 @@ _DIFFUSION_MODEL_METADATA: dict[str, DiffusionModelMetadata] = {
"Cosmos3OmniDiffusersPipeline": DiffusionModelMetadata(final_output_type="video"),
"Cosmos3OmniPipeline": DiffusionModelMetadata(final_output_type="video"),
"SanaVideoPipeline": DiffusionModelMetadata(final_output_type="video"),
"SanaImageToVideoPipeline": DiffusionModelMetadata(final_output_type="video"),
"SanaWmPipeline": DiffusionModelMetadata(
supports_multimodal_inputs=True,
max_multimodal_image_inputs=1,