Skip to content

Commit 7b76823

Browse files
committed
fix copies
1 parent 86d201d commit 7b76823

File tree

2 files changed

+14
-20
lines changed

2 files changed

+14
-20
lines changed

src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -136,35 +136,32 @@ def __init__(
136136
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.enable_vae_slicing
137137
def enable_vae_slicing(self):
138138
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.
143141
"""
144142
self.vae.enable_slicing()
145143

146144
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.disable_vae_slicing
147145
def disable_vae_slicing(self):
148146
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
150148
computing decoding in one step.
151149
"""
152150
self.vae.disable_slicing()
153151

154152
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.enable_vae_tiling
155153
def enable_vae_tiling(self):
156154
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.
161158
"""
162159
self.vae.enable_tiling()
163160

164161
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.disable_vae_tiling
165162
def disable_vae_tiling(self):
166163
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
168165
computing decoding in one step.
169166
"""
170167
self.vae.disable_tiling()

src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -143,35 +143,32 @@ def __init__(
143143
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.enable_vae_slicing
144144
def enable_vae_slicing(self):
145145
r"""
146-
Enable sliced VAE decoding.
147-
148-
When this option is enabled, the VAE will split the input tensor in slices to compute decoding in several
149-
steps. This is useful to save some memory and allow larger batch sizes.
146+
Enable sliced VAE decoding. When this option is enabled, the VAE will split the input tensor in slices to
147+
compute decoding in several steps. This is useful to save some memory and allow larger batch sizes.
150148
"""
151149
self.vae.enable_slicing()
152150

153151
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.disable_vae_slicing
154152
def disable_vae_slicing(self):
155153
r"""
156-
Disable sliced VAE decoding. If `enable_vae_slicing` was previously invoked, this method will go back to
154+
Disable sliced VAE decoding. If `enable_vae_slicing` was previously enabled, this method will go back to
157155
computing decoding in one step.
158156
"""
159157
self.vae.disable_slicing()
160158

161159
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.enable_vae_tiling
162160
def enable_vae_tiling(self):
163161
r"""
164-
Enable tiled VAE decoding.
165-
166-
When this option is enabled, the VAE will split the input tensor into tiles to compute decoding and encoding in
167-
several steps. This is useful to save a large amount of memory and to allow the processing of larger images.
162+
Enable tiled VAE decoding. When this option is enabled, the VAE will split the input tensor into tiles to
163+
compute decoding and encoding in several steps. This is useful for saving a large amount of memory and to allow
164+
processing larger images.
168165
"""
169166
self.vae.enable_tiling()
170167

171168
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.disable_vae_tiling
172169
def disable_vae_tiling(self):
173170
r"""
174-
Disable tiled VAE decoding. If `enable_vae_tiling` was previously invoked, this method will go back to
171+
Disable tiled VAE decoding. If `enable_vae_tiling` was previously enabled, this method will go back to
175172
computing decoding in one step.
176173
"""
177174
self.vae.disable_tiling()

0 commit comments

Comments
 (0)