diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py index 68342403bd14..69e2fc36fadf 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py @@ -99,7 +99,7 @@ def disable_attention_slicing(self): back to computing attention in one step. """ # set slice_size = `None` to disable `set_attention_slice` - self.enable_attention_slice(None) + self.enable_attention_slicing(None) @torch.no_grad() def __call__( diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py index 05ea84ae0326..b529598c70dc 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py @@ -118,7 +118,7 @@ def disable_attention_slicing(self): back to computing attention in one step. """ # set slice_size = `None` to disable `set_attention_slice` - self.enable_attention_slice(None) + self.enable_attention_slicing(None) @torch.no_grad() def __call__(