From d2d47c81c12060438e04635b94b8c358226b3cbc Mon Sep 17 00:00:00 2001
From: pftq <pftq@pftq.com>
Date: Mon, 17 Mar 2025 13:30:12 -0700
Subject: [PATCH] Added negative prompt option

---
 generate.py | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/generate.py b/generate.py
index 5b86324..e244203 100644
--- a/generate.py
+++ b/generate.py
@@ -136,6 +136,11 @@ def _parse_args():
         type=str,
         default=None,
         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(
         "--use_prompt_extend",
         action="store_true",
@@ -357,7 +362,9 @@ def generate(args):
                 sampling_steps=args.sample_steps,
                 guide_scale=args.sample_guide_scale,
                 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:
             if args.prompt is None:
@@ -408,13 +415,13 @@ def generate(args):
             
             # 20250227 pftq: variety batch
             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:
                     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"
+                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)
                 
             else: 
@@ -430,7 +437,9 @@ def generate(args):
                 sampling_steps=args.sample_steps,
                 guide_scale=args.sample_guide_scale,
                 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 args.save_file is None: