Update forge_loader.py
This commit is contained in:
parent
161a0938d7
commit
6020e2d7b3
@ -251,9 +251,13 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None):
|
||||
original_forward = sd_model.model.diffusion_model.forward
|
||||
|
||||
def forge_unet_forward(*args, **kwargs):
|
||||
current_transformer_options = kwargs.get('transformer_options', {})
|
||||
current_transformer_options.update(dict(cond_or_uncond=sd_model.cond_or_uncond))
|
||||
current_transformer_options.update(sd_model.forge_objects.unet.model_options.get('transformer_options', {}))
|
||||
|
||||
kwargs.update(dict(
|
||||
control=sd_model.current_controlnet_signals,
|
||||
transformer_options=sd_model.forge_objects.unet.model_options.get('transformer_options', {})
|
||||
transformer_options=current_transformer_options
|
||||
))
|
||||
return original_forward(*args, **kwargs)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user