From a313a093781b4968f7ab1f2b1902ee77b7b7956f Mon Sep 17 00:00:00 2001
From: lllyasviel <lyuminzhang@outlook.com>
Date: Sat, 27 Jan 2024 13:58:32 -0800
Subject: [PATCH] Update prompt_parser.py

---
 modules/prompt_parser.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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):