mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-06-04 21:05:48 +08:00
fix guidance cannot be none
This commit is contained in:
parent
12a02b88ac
commit
336b587fc0
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user