Commit 258c584b authored by Stefy Lanza (nextime / spora )'s avatar Stefy Lanza (nextime / spora )

Merge branch 'experimental'

parents c8cf5eac b937667b
......@@ -10126,7 +10126,7 @@ def main(args):
# 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']
i2v_pipelines = ['StableVideoDiffusionPipeline', 'I2VGenXLPipeline', 'LTXImageToVideoPipeline', 'WanPipeline', 'WanImageToVideoPipeline']
if pipeline_class_name in i2v_pipelines:
video_kwargs["image"] = init_image
......
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