diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-04 13:54:31 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-04 13:54:31 +0000 |
| commit | 52cef36f6ba169a8e606ecdcaed73d47378f0e8e (patch) | |
| tree | e14eaea39333d874f1d6ee9ed5968b7b68792425 /modules/processing.py | |
| parent | e6cbef785ab45942d0ace7f91b039c0a0d640a4a (diff) | |
| download | stable-diffusion-webui-gfx803-52cef36f6ba169a8e606ecdcaed73d47378f0e8e.tar.gz stable-diffusion-webui-gfx803-52cef36f6ba169a8e606ecdcaed73d47378f0e8e.tar.bz2 stable-diffusion-webui-gfx803-52cef36f6ba169a8e606ecdcaed73d47378f0e8e.zip | |
emergency fix for img2img
Diffstat (limited to 'modules/processing.py')
| -rw-r--r-- | modules/processing.py | 3 |
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
|