Skip to content

Commit 83233e0

Browse files
Attention slicing (huggingface#407)
uup
1 parent d5a8282 commit 83233e0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pipelines/stable_diffusion/pipeline_stable_diffusion.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,8 +87,8 @@ def disable_attention_slicing(self):
8787
Disable sliced attention computation. If `enable_attention_slicing` was previously invoked, this method will go
8888
back to computing attention in one step.
8989
"""
90-
# set slice_size = `None` to disable `set_attention_slice`
91-
self.enable_attention_slice(None)
90+
# set slice_size = `None` to disable `attention slicing`
91+
self.enable_attention_slicing(None)
9292

9393
@torch.no_grad()
9494
def __call__(

0 commit comments

Comments
 (0)