From 336b587fc0f09a1e6d37d057d1e10fdfd4bd2387 Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Sun, 11 Aug 2024 17:39:57 -0700
Subject: [PATCH] fix guidance cannot be none

---
 backend/nn/flux.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/backend/nn/flux.py b/backend/nn/flux.py
index 942d9c33..097e6622 100644
--- a/backend/nn/flux.py
+++ b/backend/nn/flux.py
@@ -372,7 +372,7 @@ class IntegratedFluxTransformer2DModel(nn.Module):
         del vec
         return img
 
-    def forward(self, x, timestep, context, y, guidance, **kwargs):
+    def forward(self, x, timestep, context, y, guidance=None, **kwargs):
         bs, c, h, w = x.shape
         input_device = x.device
         input_dtype = x.dtype