Unverified Commit 5e181edd authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

Deprecate slicing and tiling methods from `DiffusionPipeline` (#12271)

* deprecate slicing from flux pipeline.

* propagate.

* tiling

* up

* up
parent 55f0b3d7
...@@ -24,6 +24,7 @@ from ...models.transformers import FluxTransformer2DModel ...@@ -24,6 +24,7 @@ from ...models.transformers import FluxTransformer2DModel
from ...schedulers import FlowMatchEulerDiscreteScheduler from ...schedulers import FlowMatchEulerDiscreteScheduler
from ...utils import ( from ...utils import (
USE_PEFT_BACKEND, USE_PEFT_BACKEND,
deprecate,
is_torch_xla_available, is_torch_xla_available,
logging, logging,
replace_example_docstring, replace_example_docstring,
...@@ -524,6 +525,12 @@ class VisualClozeGenerationPipeline( ...@@ -524,6 +525,12 @@ class VisualClozeGenerationPipeline(
Enable sliced VAE decoding. When this option is enabled, the VAE will split the input tensor in slices to Enable sliced VAE decoding. When this option is enabled, the VAE will split the input tensor in slices to
compute decoding in several steps. This is useful to save some memory and allow larger batch sizes. compute decoding in several steps. This is useful to save some memory and allow larger batch sizes.
""" """
depr_message = f"Calling `enable_vae_slicing()` on a `{self.__class__.__name__}` is deprecated and this method will be removed in a future version. Please use `pipe.vae.enable_slicing()`."
deprecate(
"enable_vae_slicing",
"0.40.0",
depr_message,
)
self.vae.enable_slicing() self.vae.enable_slicing()
def disable_vae_slicing(self): def disable_vae_slicing(self):
...@@ -531,6 +538,12 @@ class VisualClozeGenerationPipeline( ...@@ -531,6 +538,12 @@ class VisualClozeGenerationPipeline(
Disable sliced VAE decoding. If `enable_vae_slicing` was previously enabled, this method will go back to Disable sliced VAE decoding. If `enable_vae_slicing` was previously enabled, this method will go back to
computing decoding in one step. computing decoding in one step.
""" """
depr_message = f"Calling `disable_vae_slicing()` on a `{self.__class__.__name__}` is deprecated and this method will be removed in a future version. Please use `pipe.vae.disable_slicing()`."
deprecate(
"disable_vae_slicing",
"0.40.0",
depr_message,
)
self.vae.disable_slicing() self.vae.disable_slicing()
def enable_vae_tiling(self): def enable_vae_tiling(self):
...@@ -539,6 +552,12 @@ class VisualClozeGenerationPipeline( ...@@ -539,6 +552,12 @@ class VisualClozeGenerationPipeline(
compute decoding and encoding in several steps. This is useful for saving a large amount of memory and to allow compute decoding and encoding in several steps. This is useful for saving a large amount of memory and to allow
processing larger images. processing larger images.
""" """
depr_message = f"Calling `enable_vae_tiling()` on a `{self.__class__.__name__}` is deprecated and this method will be removed in a future version. Please use `pipe.vae.enable_tiling()`."
deprecate(
"enable_vae_tiling",
"0.40.0",
depr_message,
)
self.vae.enable_tiling() self.vae.enable_tiling()
def disable_vae_tiling(self): def disable_vae_tiling(self):
...@@ -546,6 +565,12 @@ class VisualClozeGenerationPipeline( ...@@ -546,6 +565,12 @@ class VisualClozeGenerationPipeline(
Disable tiled VAE decoding. If `enable_vae_tiling` was previously enabled, this method will go back to Disable tiled VAE decoding. If `enable_vae_tiling` was previously enabled, this method will go back to
computing decoding in one step. computing decoding in one step.
""" """
depr_message = f"Calling `disable_vae_tiling()` on a `{self.__class__.__name__}` is deprecated and this method will be removed in a future version. Please use `pipe.vae.disable_tiling()`."
deprecate(
"disable_vae_tiling",
"0.40.0",
depr_message,
)
self.vae.disable_tiling() self.vae.disable_tiling()
def _prepare_latents(self, image, mask, gen, vae_scale_factor, device, dtype): def _prepare_latents(self, image, mask, gen, vae_scale_factor, device, dtype):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment