diff options
author | Volpeon <git@volpeon.ink> | 2022-12-04 09:24:37 +0100 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2022-12-04 09:24:37 +0100 |
commit | 2825ba2f2030b2fd3e841aad416a4fd28d67615a (patch) | |
tree | 180afc9ae42c7d29b7229423d4db53183e6ad49b /pipelines/stable_diffusion | |
parent | Update (diff) | |
download | textual-inversion-diff-2825ba2f2030b2fd3e841aad416a4fd28d67615a.tar.gz textual-inversion-diff-2825ba2f2030b2fd3e841aad416a4fd28d67615a.tar.bz2 textual-inversion-diff-2825ba2f2030b2fd3e841aad416a4fd28d67615a.zip |
Update
Diffstat (limited to 'pipelines/stable_diffusion')
-rw-r--r-- | pipelines/stable_diffusion/vlpn_stable_diffusion.py | 16 |
1 files changed, 0 insertions, 16 deletions
diff --git a/pipelines/stable_diffusion/vlpn_stable_diffusion.py b/pipelines/stable_diffusion/vlpn_stable_diffusion.py index 9b51763..f80e951 100644 --- a/pipelines/stable_diffusion/vlpn_stable_diffusion.py +++ b/pipelines/stable_diffusion/vlpn_stable_diffusion.py | |||
@@ -78,22 +78,6 @@ class VlpnStableDiffusion(DiffusionPipeline): | |||
78 | scheduler=scheduler, | 78 | scheduler=scheduler, |
79 | ) | 79 | ) |
80 | 80 | ||
81 | def enable_xformers_memory_efficient_attention(self): | ||
82 | r""" | ||
83 | Enable memory efficient attention as implemented in xformers. | ||
84 | When this option is enabled, you should observe lower GPU memory usage and a potential speed up at inference | ||
85 | time. Speed up at training time is not guaranteed. | ||
86 | Warning: When Memory Efficient Attention and Sliced attention are both enabled, the Memory Efficient Attention | ||
87 | is used. | ||
88 | """ | ||
89 | self.unet.set_use_memory_efficient_attention_xformers(True) | ||
90 | |||
91 | def disable_xformers_memory_efficient_attention(self): | ||
92 | r""" | ||
93 | Disable memory efficient attention as implemented in xformers. | ||
94 | """ | ||
95 | self.unet.set_use_memory_efficient_attention_xformers(False) | ||
96 | |||
97 | def enable_attention_slicing(self, slice_size: Optional[Union[str, int]] = "auto"): | 81 | def enable_attention_slicing(self, slice_size: Optional[Union[str, int]] = "auto"): |
98 | r""" | 82 | r""" |
99 | Enable sliced attention computation. | 83 | Enable sliced attention computation. |