add nvidia-chronoedit

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-10-30 19:50:02 -04:00
parent 76d699dc09
commit 3ae10dd0e1
20 changed files with 1371 additions and 22 deletions
+5 -2
View File
@@ -138,8 +138,11 @@ def task_specific_kwargs(p, model):
'width': p.width,
'height': p.height,
}
if ('WanImageToVideoPipeline' in model_cls) and (p.init_images is not None) and (len(p.init_images) > 0):
task_args['image'] = p.init_images[0]
if ('WanImageToVideoPipeline' in model_cls) or ('ChronoEditPipeline' in model_cls):
if (p.init_images is not None) and (len(p.init_images) > 0):
task_args['image'] = p.init_images[0]
else:
task_args['image'] = Image.new('RGB', (p.width, p.height), (0, 0, 0)) # monkey-patch so wan-i2i pipeline does not error-out on t2i
if ('WanVACEPipeline' in model_cls) and (p.init_images is not None) and (len(p.init_images) > 0):
task_args['reference_images'] = p.init_images
if 'BlipDiffusionPipeline' in model_cls: