fix guidance cannot be none

This commit is contained in:
layerdiffusion
2024-08-11 17:39:57 -07:00
parent 12a02b88ac
commit 336b587fc0

View File

@@ -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