From 8ed8bf52d092326b532fac6d6f9db3f66f4f0291 Mon Sep 17 00:00:00 2001
From: damian <null@damianstewart.com>
Date: Thu, 26 Jan 2023 17:04:22 +0100
Subject: [PATCH] use 'auto' slice size
---
ldm/invoke/generator/diffusers_pipeline.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/ldm/invoke/generator/diffusers_pipeline.py b/ldm/invoke/generator/diffusers_pipeline.py
index a16cbe594b..b7ad925c8c 100644
--- a/ldm/invoke/generator/diffusers_pipeline.py
+++ b/ldm/invoke/generator/diffusers_pipeline.py
@@ -313,8 +313,7 @@ class StableDiffusionGeneratorPipeline(StableDiffusionPipeline):
# fix is in https://github.com/kulinseth/pytorch/pull/222 but no idea when it will get merged to pytorch mainline.
pass
else:
- slice_size = 4 # or 2, or 8. i chose this arbitrarily.
- self.enable_attention_slicing(slice_size=slice_size)
+ self.enable_attention_slicing(slice_size='auto')
def image_from_embeddings(self, latents: torch.Tensor, num_inference_steps: int,