diff options
-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
|