diff --git a/modules/prompt_parser.py b/modules/prompt_parser.py index 0c86c356..ae642529 100644 --- a/modules/prompt_parser.py +++ b/modules/prompt_parser.py @@ -338,7 +338,7 @@ def stack_conds_alter(tensors, weights): full_weights = full_weights + float(w) result = result / full_weights - return result + return result[None] def reconstruct_multicond_batch(c: MulticondLearnedConditioning, current_step):