Skip to content

Commit 3097441

Browse files
committed
add copy statements
1 parent 9d5c0b5 commit 3097441

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/diffusers/pipelines/sana/pipeline_sana_sprint_img2img.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -170,23 +170,23 @@ def __init__(
170170
)
171171
self.image_processor = PixArtImageProcessor(vae_scale_factor=self.vae_scale_factor)
172172

173-
# Copied from diffusers.pipelines.sana.pipeline_sana.enable_vae_slicing
173+
# Copied from diffusers.pipelines.sana.pipeline_sana.SanaPipeline.enable_vae_slicing
174174
def enable_vae_slicing(self):
175175
r"""
176176
Enable sliced VAE decoding. When this option is enabled, the VAE will split the input tensor in slices to
177177
compute decoding in several steps. This is useful to save some memory and allow larger batch sizes.
178178
"""
179179
self.vae.enable_slicing()
180180

181-
# Copied from diffusers.pipelines.sana.pipeline_sana.disable_vae_slicing
181+
# Copied from diffusers.pipelines.sana.pipeline_sana.SanaPipeline.disable_vae_slicing
182182
def disable_vae_slicing(self):
183183
r"""
184184
Disable sliced VAE decoding. If `enable_vae_slicing` was previously enabled, this method will go back to
185185
computing decoding in one step.
186186
"""
187187
self.vae.disable_slicing()
188188

189-
# Copied from diffusers.pipelines.sana.pipeline_sana.enable_vae_tiling
189+
# Copied from diffusers.pipelines.sana.pipeline_sana.SanaPipeline.enable_vae_tiling
190190
def enable_vae_tiling(self):
191191
r"""
192192
Enable tiled VAE decoding. When this option is enabled, the VAE will split the input tensor into tiles to

0 commit comments

Comments
 (0)