Fixed READ_WRITE state for ContextRef behaving like READ state instead

This commit is contained in:
Jedrzej Kosinski
2024-07-31 21:04:52 -05:00
parent f1916a4a0b
commit f25e35f1fe
+2 -2
View File
@@ -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