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

Added optional --variety_batch flag to vary the CFG and Steps within a batch.

Browse Source
This commit is contained in:
pftq 2025-02-27 10:37:55 -08:00 committed by GitHub
parent 5bd54ebe00
commit 3aec13addb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 22 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
generate.py
View File

@ -194,6 +194,13 @@ def _parse_args():
default=1,
help="Number of videos per run without reloading the model again.")
# 20250227 pftq: variety batch with different CFG/steps each video, labeled in filename.
parser.add_argument(
"--variety_batch",
action="store_true",
default=False,
help="Different CFG/steps each video, labeled in filename.")
args = parser.parse_args()
_validate_args(args)
@ -398,7 +405,21 @@ def generate(args):
t5_cpu=args.t5_cpu,
)
logging.info("Generating video...")
# 20250227 pftq: variety batch
if args.variety_batch and batch_index > 0:
args.sample_guide_scale = args.sample_guide_scale + 0.5
if args.sample_guide_scale > 10:
args.sample_guide_scale = 3
args.sample_steps = args.sample_steps + 10
if args.sample_steps > 150:
args.sample_stemps = 50
args.save_file = datetime.now().strftime("%Y-%m-%d_%H-%M-%S") + "_wan-server_CFG-"+str(args.sample_guide_scale)+"_steps-"+str(args.sample_steps)+"_"+str(args.ulysses_size)+"-GPUs-"+str(args.frame_num)+"f.mp4"
logging.info("Generating video... "+args.save_file)
else:
logging.info("Generating video...")
video = wan_i2v.generate(
args.prompt,
img,