use better context manager to fix potential problems

2026-02-04 06:59:59 +00:00 · 2024-02-08 01:51:18 -08:00
parent 4c9db26541
commit 760f727eb9
2 changed files with 4 additions and 1 deletions
--- 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)