Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "b30f3cdba32b9e4b577e66536dad111d6020a3fa"
Unverified Commit 8ff777d3 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

Attention slicing (#407) 

uup
parent 1a431ae8
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py
View file @ 8ff777d3
...@@ -87,8 +87,8 @@ class StableDiffusionPipeline(DiffusionPipeline): ...@@ -87,8 +87,8 @@ class StableDiffusionPipeline(DiffusionPipeline):
Disable sliced attention computation. If `enable_attention_slicing` was previously invoked, this method will go Disable sliced attention computation. If `enable_attention_slicing` was previously invoked, this method will go
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 `attention slicing`
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