cxllm-vendors/stable-diffusion-webui-forge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update forge_loader.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 19:05:00 -08:00
parent 161a0938d7
commit 6020e2d7b3
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules_forge/forge_loader.py
+5 -1
View File
@@ -251,9 +251,13 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None):
original_forward = sd_model.model.diffusion_model.forward original_forward = sd_model.model.diffusion_model.forward
def forge_unet_forward(*args, **kwargs): 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( kwargs.update(dict(
control=sd_model.current_controlnet_signals, 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) return original_forward(*args, **kwargs)