diff --git a/modules/processing.py b/modules/processing.py
index 7ea8ad4b..a23607e7 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.no_grad():
+    with torch.inference_mode():
         with devices.autocast():
             p.init(p.all_prompts, p.all_seeds, p.all_subseeds)