Skip to content

Typo in StableDiffusionInpaintPipeline.disable_attention_slicing() method  #497

@juno-hwang

Description

@juno-hwang

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

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions