Update generate.py

pull/729/head
hithereai 2023-05-19 19:19:12 +03:00
parent 6cdfd48943
commit c73dcf58da
1 changed files with 1 additions and 1 deletions

View File

@ -246,7 +246,7 @@ def generate_inner(args, keys, anim_args, loop_args, controlnet_args, root, fram
mask = None mask = None
assert not ((mask is not None and args.use_mask and args.overlay_mask) and ( assert not ((mask is not None and args.use_mask and args.overlay_mask) and (
args.init_sample is None and init_image is None)), "Need an init image when use_mask == True and overlay_mask == True" args.init_sample is None and init_image is None)), "Need an init image when use_mask == True and overlay_mask == True"
p.init_images = [init_image] p.init_images = [init_image]
p.image_mask = mask p.image_mask = mask