Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-06-18 13:38:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Compare commits

base: Zengtudor:1a55891718947978c1acd6c5da037ea5df739066
Branches Tags
Zengtudor:main
Zengtudor:dev/format
Zengtudor:dev/vace_seed
Zengtudor:dev/vace_dev
Zengtudor:dev/vace_size
Zengtudor:dev/vace
..
compare: Zengtudor:20e1bc2503313149b3286f1fd92c7eca40f2fbdc
Branches Tags
Zengtudor:main
Zengtudor:dev/format
Zengtudor:dev/vace_seed
Zengtudor:dev/vace_dev
Zengtudor:dev/vace_size
Zengtudor:dev/vace

No commits in common. "1a55891718947978c1acd6c5da037ea5df739066" and "20e1bc2503313149b3286f1fd92c7eca40f2fbdc" have entirely different histories.
1a55891718 ... 20e1bc2503

1 changed files with 7 additions and 3 deletions
Show Stats Expand all files Collapse all files

10
generate.py
View File

@ -61,7 +61,9 @@ def _validate_args(args):
# The default sampling steps are 40 for image-to-video tasks and 50 for text-to-video tasks. # The default sampling steps are 40 for image-to-video tasks and 50 for text-to-video tasks.
if args.sample_steps is None: if args.sample_steps is None:
args.sample_steps = 50 args.sample_steps = 50
if "i2v" in args.task: if "vace" in args.task:
args.sample_steps = 25
elif "i2v" in args.task:
args.sample_steps = 40 args.sample_steps = 40
@ -69,8 +71,10 @@ def _validate_args(args):
args.sample_shift = 5.0 args.sample_shift = 5.0
if "i2v" in args.task and args.size in ["832*480", "480*832"]: if "i2v" in args.task and args.size in ["832*480", "480*832"]:
args.sample_shift = 3.0 args.sample_shift = 3.0
elif "flf2v" in args.task or "vace" in args.task: elif "flf2v" in args.task:
args.sample_shift = 16 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. # The default number of frames are 1 for text-to-image tasks and 81 for other tasks.
@ -505,7 +509,7 @@ def generate(args):
args.prompt = input_prompt[0] args.prompt = input_prompt[0]
logging.info(f"Extended prompt: {args.prompt}") logging.info(f"Extended prompt: {args.prompt}")
logging.info("Creating VACE pipeline.") logging.info("Creating WanT2V pipeline.")
wan_vace = wan.WanVace( wan_vace = wan.WanVace(
config=cfg, config=cfg,
checkpoint_dir=args.ckpt_dir, checkpoint_dir=args.ckpt_dir,