diff --git a/tests/entrypoints/openai_api/test_video_pipeline_capability.py b/tests/entrypoints/openai_api/test_video_pipeline_capability.py index cfa11f96d..985aadf89 100644 --- a/tests/entrypoints/openai_api/test_video_pipeline_capability.py +++ b/tests/entrypoints/openai_api/test_video_pipeline_capability.py @@ -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" diff --git a/vllm_omni/diffusion/model_metadata.py b/vllm_omni/diffusion/model_metadata.py index 1665d6a16..7f15955ec 100644 --- a/vllm_omni/diffusion/model_metadata.py +++ b/vllm_omni/diffusion/model_metadata.py @@ -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,