From 57cc24dca8aeae5962cc9e3e6852a2a8310daaea Mon Sep 17 00:00:00 2001
From: Raphael Walker <slickytail.mc@gmail.com>
Date: Thu, 5 Dec 2024 16:05:15 +0100
Subject: [PATCH] arg names

---
 comfy_extras/nodes_flux.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy_extras/nodes_flux.py b/comfy_extras/nodes_flux.py
index 307206cfb..4da26e565 100644
--- a/comfy_extras/nodes_flux.py
+++ b/comfy_extras/nodes_flux.py
@@ -72,7 +72,7 @@ class _ReduxAttnWrapper:
                 x, c = self.previous(img=args["img"], txt=args["txt"],vec=args["vec"], pe=args["pe"], attn_mask=mask)
                 return {"img": x, "txt": c}
             case SingleStreamBlock():
-                x = self.previous(img=args["img"], vec=args["vec"], pe=args["pe"], attn_mask=mask)
+                x = self.previous(args["img"], vec=args["vec"], pe=args["pe"], attn_mask=mask)
                 return {"img": x}
             case _ReduxAttnWrapper():
                 # pass along the mask, and tell the next redux what its part of the mask is