From 71bbd78574ed6d506658c6e3a73ba5628ffc126a Mon Sep 17 00:00:00 2001 From: Jonathan <34005131+JPPhoto@users.noreply.github.com> Date: Tue, 21 Feb 2023 12:35:03 -0600 Subject: [PATCH] Fix crashing when using 2.1 model We now require more free memory to avoid attention slicing. 17.5% free was not sufficient headroom, so now we require 25%. --- ldm/invoke/generator/diffusers_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ldm/invoke/generator/diffusers_pipeline.py b/ldm/invoke/generator/diffusers_pipeline.py index 6755bbb880..012301252e 100644 --- a/ldm/invoke/generator/diffusers_pipeline.py +++ b/ldm/invoke/generator/diffusers_pipeline.py @@ -330,7 +330,7 @@ class StableDiffusionGeneratorPipeline(StableDiffusionPipeline): 16 * \ latents.size(dim=2) * latents.size(dim=3) * latents.size(dim=2) * latents.size(dim=3) * \ bytes_per_element_needed_for_baddbmm_duplication - if max_size_required_for_baddbmm > (mem_free * 3.3 / 4.0): # 3.3 / 4.0 is from old Invoke code + if max_size_required_for_baddbmm > (mem_free * 3.0 / 4.0): # 3.3 / 4.0 is from old Invoke code self.enable_attention_slicing(slice_size='max') else: self.disable_attention_slicing()