From 7dd4265cc03ef2fedf1935c17b7a9c74ae5fba0b Mon Sep 17 00:00:00 2001
From: drhead <1313496+drhead@users.noreply.github.com>
Date: Fri, 23 May 2025 18:13:44 -0400
Subject: [PATCH] Add preview-stream cli arg

---
 comfy/cli_args.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/comfy/cli_args.py b/comfy/cli_args.py
index 4fb675f99..e25b00e46 100644
--- a/comfy/cli_args.py
+++ b/comfy/cli_args.py
@@ -100,6 +100,8 @@ parser.add_argument("--preview-method", type=LatentPreviewMethod, default=Latent
 
 parser.add_argument("--preview-size", type=int, default=512, help="Sets the maximum preview size for sampler nodes.")
 
+parser.add_argument("--preview-stream", action="store_true", help="Use a CUDA Stream to reduce performance cost of previews.")
+
 cache_group = parser.add_mutually_exclusive_group()
 cache_group.add_argument("--cache-classic", action="store_true", help="Use the old style (aggressive) caching.")
 cache_group.add_argument("--cache-lru", type=int, default=0, help="Use LRU caching with a maximum of N node results cached. May use more RAM/VRAM.")