Unverified Commit f7cd6b87 authored by Pedro Cuenca's avatar Pedro Cuenca Committed by GitHub
Browse files

Fix `disable_attention_slicing` in pipelines (#498)

Fix `disable_attention_slicing` in pipelines.
parent 721e0174
...@@ -99,7 +99,7 @@ class StableDiffusionImg2ImgPipeline(DiffusionPipeline): ...@@ -99,7 +99,7 @@ class StableDiffusionImg2ImgPipeline(DiffusionPipeline):
back to computing attention in one step. back to computing attention in one step.
""" """
# set slice_size = `None` to disable `set_attention_slice` # set slice_size = `None` to disable `set_attention_slice`
self.enable_attention_slice(None) self.enable_attention_slicing(None)
@torch.no_grad() @torch.no_grad()
def __call__( def __call__(
......
...@@ -118,7 +118,7 @@ class StableDiffusionInpaintPipeline(DiffusionPipeline): ...@@ -118,7 +118,7 @@ class StableDiffusionInpaintPipeline(DiffusionPipeline):
back to computing attention in one step. back to computing attention in one step.
""" """
# set slice_size = `None` to disable `set_attention_slice` # set slice_size = `None` to disable `set_attention_slice`
self.enable_attention_slice(None) self.enable_attention_slicing(None)
@torch.no_grad() @torch.no_grad()
def __call__( def __call__(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment