aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2022-10-04 13:54:31 +0000
committerAUTOMATIC <16777216c@gmail.com>2022-10-04 13:54:31 +0000
commit52cef36f6ba169a8e606ecdcaed73d47378f0e8e (patch)
treee14eaea39333d874f1d6ee9ed5968b7b68792425
parente6cbef785ab45942d0ace7f91b039c0a0d640a4a (diff)
downloadstable-diffusion-webui-gfx803-52cef36f6ba169a8e606ecdcaed73d47378f0e8e.tar.gz
stable-diffusion-webui-gfx803-52cef36f6ba169a8e606ecdcaed73d47378f0e8e.tar.bz2
stable-diffusion-webui-gfx803-52cef36f6ba169a8e606ecdcaed73d47378f0e8e.zip
emergency fix for img2img
-rw-r--r--modules/processing.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/processing.py b/modules/processing.py
index 6f5599c7..e9c45394 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -331,7 +331,8 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
output_images = []
with torch.no_grad():
- p.init(all_prompts, all_seeds, all_subseeds)
+ with devices.autocast():
+ p.init(all_prompts, all_seeds, all_subseeds)
if state.job_count == -1:
state.job_count = p.n_iter