forgot arg

demo_watermark.py

@@ -154,6 +154,12 @@ def parse_args():
         default=False,
         help="Skip the model loading to debug the interface.",
     )
+    parser.add_argument(
+        "--seed_separately",
+        type=str2bool,
+        default=True,
+        help="Whether to call the torch seed function before both the unwatermarked and watermarked generate calls.",
+    )
     args = parser.parse_args()
     return args