Unverified Commit 8ff777d3 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

Attention slicing (#407)

uup
parent 1a431ae8
......@@ -87,8 +87,8 @@ class StableDiffusionPipeline(DiffusionPipeline):
Disable sliced attention computation. If `enable_attention_slicing` was previously invoked, this method will go
back to computing attention in one step.
"""
# set slice_size = `None` to disable `set_attention_slice`
self.enable_attention_slice(None)
# set slice_size = `None` to disable `attention slicing`
self.enable_attention_slicing(None)
@torch.no_grad()
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