dont use index_in directly

ljleb · Feb 2, 2024 · 9420c78 · 9420c78
1 parent 185638f
commit 9420c78
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)