From 3db3dc30446ead4ab5999429ee49e9505887e7d1 Mon Sep 17 00:00:00 2001
From: RyanOnTheInside <7623207+ryanontheinside@users.noreply.github.com>
Date: Sat, 8 Feb 2025 17:47:23 -0500
Subject: [PATCH] dynamic buffer

---
 stream_diffusion_nodes.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/stream_diffusion_nodes.py b/stream_diffusion_nodes.py
index 9d766fc..03db7dd 100644
--- a/stream_diffusion_nodes.py
+++ b/stream_diffusion_nodes.py
@@ -74,12 +74,19 @@ class StreamDiffusionPipeline(ControlNodeBase):
             "sigmas": None,
             "stock_noise": None,  # For RCFG
             "timesteps": None,
+            "current_buffer_size": None,  # Track buffer size
         })
         
         # Get latent tensor from dict
         samples = latents["samples"]  # Shape: [1, 4, H, W]
         device = samples.device
         
+        # Check if buffer size changed
+        if state["current_buffer_size"] != frame_buffer_size:
+            state["latent_buffer"] = None  # Force reinitialization
+            state["current_buffer_size"] = frame_buffer_size
+            state["current_idx"] = 0
+        
         # Initialize scheduler and timesteps if needed
         if state["sigmas"] is None:
             # Calculate sigmas using ComfyUI's scheduler system