dont use index_in directly

2024-02-02 12:13:22 -05:00 · 2024-02-02 12:13:22 -05:00 · 9420c7871d
parent 185638f26e
commit 9420c7871d
1 changed files with 1 additions and 1 deletions
--- a/lib_neutral_prompt/cfg_denoiser_hijack.py
+++ b/lib_neutral_prompt/cfg_denoiser_hijack.py
@ -84,7 +84,7 @@ def gather_webui_conds(
    for child in prompt.children:
        if child.conciliation is None:
            if isinstance(child, neutral_prompt_parser.LeafPrompt):
-                child_x_out = args.x_out[index_in]
+                child_x_out = args.x_out[args.cond_indices[index_in][0]]
            else:
                child_x_out = child.accept(CondDeltaVisitor(), args, index_in)
                child_x_out += child.accept(AuxCondDeltaVisitor(), args, child_x_out, index_in)