Update vace args

This commit is contained in:
wangang.wa 2025-05-14 10:02:57 +08:00
parent 20e1bc2503
commit a6271b1011

View File

@ -61,9 +61,7 @@ def _validate_args(args):
# The default sampling steps are 40 for image-to-video tasks and 50 for text-to-video tasks.
if args.sample_steps is None:
args.sample_steps = 50
if "vace" in args.task:
args.sample_steps = 25
elif "i2v" in args.task:
if "i2v" in args.task:
args.sample_steps = 40
@ -71,10 +69,8 @@ def _validate_args(args):
args.sample_shift = 5.0
if "i2v" in args.task and args.size in ["832*480", "480*832"]:
args.sample_shift = 3.0
elif "flf2v" in args.task:
elif "flf2v" in args.task or "vace" in args.task:
args.sample_shift = 16
elif "vace" in args.task:
args.sample_shift = 8.0
# The default number of frames are 1 for text-to-image tasks and 81 for other tasks.