Update forge_loader.py

This commit is contained in:
lllyasviel
2024-01-25 19:05:11 -08:00
parent 6020e2d7b3
commit 6d9b9707bc

View File

@@ -254,7 +254,7 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None):
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=current_transformer_options