From f25e35f1fe299f57e9c7980b36e4d96809543c43 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Wed, 31 Jul 2024 21:04:52 -0500 Subject: [PATCH] Fixed READ_WRITE state for ContextRef behaving like READ state instead --- adv_control/control_reference.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/adv_control/control_reference.py b/adv_control/control_reference.py index 84f7973..1fcb5d0 100644 --- a/adv_control/control_reference.py +++ b/adv_control/control_reference.py @@ -652,12 +652,12 @@ def factory_forward_inject_UNetModel(reference_injections: ReferenceInjections): # attn if is_read(transformer_options[CONTEXTREF_MACHINE_STATE]): read_attn_list.extend(context_attn_controlnets) - elif is_write(transformer_options[CONTEXTREF_MACHINE_STATE]): + if is_write(transformer_options[CONTEXTREF_MACHINE_STATE]): write_attn_list.extend(context_attn_controlnets) # adain if is_read(transformer_options[CONTEXTREF_MACHINE_STATE]): read_adain_list.extend(context_adain_controlnets) - elif is_write(transformer_options[CONTEXTREF_MACHINE_STATE]): + if is_write(transformer_options[CONTEXTREF_MACHINE_STATE]): write_adain_list.extend(context_adain_controlnets) # apply lists, containing both RefCN and ContextRef transformer_options[REF_READ_ATTN_CONTROL_LIST] = read_attn_list