@@ -136,35 +136,32 @@ def __init__(
136
136
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.enable_vae_slicing
137
137
def enable_vae_slicing (self ):
138
138
r"""
139
- Enable sliced VAE decoding.
140
-
141
- When this option is enabled, the VAE will split the input tensor in slices to compute decoding in several
142
- steps. This is useful to save some memory and allow larger batch sizes.
139
+ Enable sliced VAE decoding. When this option is enabled, the VAE will split the input tensor in slices to
140
+ compute decoding in several steps. This is useful to save some memory and allow larger batch sizes.
143
141
"""
144
142
self .vae .enable_slicing ()
145
143
146
144
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.disable_vae_slicing
147
145
def disable_vae_slicing (self ):
148
146
r"""
149
- Disable sliced VAE decoding. If `enable_vae_slicing` was previously invoked , this method will go back to
147
+ Disable sliced VAE decoding. If `enable_vae_slicing` was previously enabled , this method will go back to
150
148
computing decoding in one step.
151
149
"""
152
150
self .vae .disable_slicing ()
153
151
154
152
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.enable_vae_tiling
155
153
def enable_vae_tiling (self ):
156
154
r"""
157
- Enable tiled VAE decoding.
158
-
159
- When this option is enabled, the VAE will split the input tensor into tiles to compute decoding and encoding in
160
- several steps. This is useful to save a large amount of memory and to allow the processing of larger images.
155
+ Enable tiled VAE decoding. When this option is enabled, the VAE will split the input tensor into tiles to
156
+ compute decoding and encoding in several steps. This is useful for saving a large amount of memory and to allow
157
+ processing larger images.
161
158
"""
162
159
self .vae .enable_tiling ()
163
160
164
161
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.disable_vae_tiling
165
162
def disable_vae_tiling (self ):
166
163
r"""
167
- Disable tiled VAE decoding. If `enable_vae_tiling` was previously invoked , this method will go back to
164
+ Disable tiled VAE decoding. If `enable_vae_tiling` was previously enabled , this method will go back to
168
165
computing decoding in one step.
169
166
"""
170
167
self .vae .disable_tiling ()
0 commit comments