From 16caff3d142fa87003f4a8c4e501553833abeb7d Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Fri, 23 Feb 2024 16:13:40 -0800 Subject: [PATCH] Revert "try solve #381" This reverts commit 9d4c88912d808b9d8eba0c78cf332a712d3cc16f. --- modules/processing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)