From 9d4c88912d808b9d8eba0c78cf332a712d3cc16f Mon Sep 17 00:00:00 2001
From: lllyasviel <lyuminzhang@outlook.com>
Date: Fri, 23 Feb 2024 16:01:40 -0800
Subject: [PATCH] try solve #381

caused by some potential historical webui problems
---
 modules/processing.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/processing.py b/modules/processing.py
index a23607e7..7ea8ad4b 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -814,7 +814,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
 
     infotexts = []
     output_images = []
-    with torch.inference_mode():
+    with torch.no_grad():
         with devices.autocast():
             p.init(p.all_prompts, p.all_seeds, p.all_subseeds)