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 negative prompt option

Browse Source
This commit is contained in:
pftq 2025-03-17 13:30:12 -07:00 committed by GitHub
parent 68db072ae0
commit d2d47c81c1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
generate.py
View File

@ -136,6 +136,11 @@ def _parse_args():
type=str, type=str,
default=None, default=None,
help="The prompt to generate the image or video from.") help="The prompt to generate the image or video from.")
parser.add_argument(
"--negative_prompt",
type=str,
default=None,
help="Optional negative prompt, defaults employed if not set.")
parser.add_argument( parser.add_argument(
"--use_prompt_extend", "--use_prompt_extend",
action="store_true", action="store_true",
@ -357,7 +362,9 @@ def generate(args):
sampling_steps=args.sample_steps, sampling_steps=args.sample_steps,
guide_scale=args.sample_guide_scale, guide_scale=args.sample_guide_scale,
seed=args.base_seed, seed=args.base_seed,
offload_model=args.offload_model) offload_model=args.offload_model,
n_prompt=args.negative_prompt # 20250316 pftq: Optional negative prompt
)
else: else:
if args.prompt is None: if args.prompt is None:
@ -408,13 +415,13 @@ def generate(args):
# 20250227 pftq: variety batch # 20250227 pftq: variety batch
if args.variety_batch and batch_index > 0: if args.variety_batch and batch_index > 0:
args.sample_guide_scale = args.sample_guide_scale + 0.5 args.sample_guide_scale = args.sample_guide_scale + 1
if args.sample_guide_scale > 10: if args.sample_guide_scale > 10:
args.sample_guide_scale = 3 args.sample_guide_scale = 3
args.sample_steps = args.sample_steps + 10 args.sample_steps = args.sample_steps + 10
if args.sample_steps > 150: if args.sample_steps > 150:
args.sample_stemps = 50 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" args.save_file = datetime.now().strftime("%Y-%m-%d_%H-%M-%S") + "_wan-server_"+str(args.frame_num)+"f_CFG-"+str(args.sample_guide_scale)+"_steps-"+str(args.sample_steps)+"_"+str(args.ulysses_size)+"-GPUs.mp4"
logging.info("Generating video... "+args.save_file) logging.info("Generating video... "+args.save_file)
else: else:
@ -430,7 +437,9 @@ def generate(args):
sampling_steps=args.sample_steps, sampling_steps=args.sample_steps,
guide_scale=args.sample_guide_scale, guide_scale=args.sample_guide_scale,
seed=args.base_seed, seed=args.base_seed,
offload_model=args.offload_model) offload_model=args.offload_model,
n_prompt=args.negative_prompt # 20250316 pftq: Optional negative prompt
)
if rank == 0: if rank == 0:
if args.save_file is None: if args.save_file is None: