-
Notifications
You must be signed in to change notification settings - Fork 6.6k
Closed
Labels
bugSomething isn't workingSomething isn't working
Description
Describe the bug
To enable/disable attention slicing, one need to use enable/disable_attention_slicing() method.
but in StableDiffusionInpaintPipeline class, there is misspelling in disable_attention_slicing() method.
self.enable_attention_slice(None) should be replaced as self.enable_attention_slicing(None).
Reproduction
No response
Logs
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
<ipython-input-5-cdc2e7fc6066> in <module>
34 else:
35 pipe.disable_attention_slicing()
---> 36 pipe_inpaint.disable_attention_slicing()
37
38 for i in range(10):
/usr/local/lib/python3.7/dist-packages/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py in disable_attention_slicing(self)
119 """
120 # set slice_size = `None` to disable `set_attention_slice`
--> 121 self.enable_attention_slice(None)
122
123 @torch.no_grad()
AttributeError: 'StableDiffusionInpaintPipeline' object has no attribute 'enable_attention_slice'System Info
Google Colab Pro+ ( with V100 gpu )
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't working