Fix I2V check to verify pipeline class supports image input

- LTXPipeline is T2V-only, not I2V-capable
- Check both supports_i2v flag AND pipeline class name
- I2V-capable pipelines: StableVideoDiffusionPipeline, I2VGenXLPipeline, LTXImageToVideoPipeline, WanPipeline
- Prevents passing 'image' argument to T2V-only pipelines like LTXPipeline
parent 0af33f9c
......@@ -8843,10 +8843,16 @@ def main(args):
}
if (args.image_to_video or args.image) and init_image is not None:
if m_info.get("supports_i2v"):
# Check if the pipeline class actually supports image input
# I2V-capable pipelines: StableVideoDiffusionPipeline, I2VGenXLPipeline, LTXImageToVideoPipeline, WanPipeline (with I2V variant)
# T2V-only pipelines: LTXPipeline, MochiPipeline, CogVideoXPipeline, etc.
pipeline_class_name = PipelineClass.__name__ if hasattr(PipelineClass, '__name__') else str(PipelineClass)
i2v_pipelines = ['StableVideoDiffusionPipeline', 'I2VGenXLPipeline', 'LTXImageToVideoPipeline', 'WanPipeline']
if m_info.get("supports_i2v") and pipeline_class_name in i2v_pipelines:
video_kwargs["image"] = init_image
else:
print(f"Warning: {args.model} does not support 'image' argument – running pure T2V")
print(f"Warning: {args.model} ({pipeline_class_name}) does not support 'image' argument – running pure T2V")
output = pipe(**video_kwargs)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment