Skip to content

Commit d34b18c

Browse files
DN6sayakpaul
andauthored
Deprecate Stable Cascade (huggingface#12537)
* update * update --------- Co-authored-by: Sayak Paul <[email protected]>
1 parent 7536f64 commit d34b18c

File tree

3 files changed

+12
-6
lines changed

3 files changed

+12
-6
lines changed

src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
from ...schedulers import DDPMWuerstchenScheduler
2222
from ...utils import is_torch_version, is_torch_xla_available, logging, replace_example_docstring
2323
from ...utils.torch_utils import randn_tensor
24-
from ..pipeline_utils import DiffusionPipeline, ImagePipelineOutput
24+
from ..pipeline_utils import DeprecatedPipelineMixin, DiffusionPipeline, ImagePipelineOutput
2525
from ..wuerstchen.modeling_paella_vq_model import PaellaVQModel
2626

2727

@@ -55,7 +55,7 @@
5555
"""
5656

5757

58-
class StableCascadeDecoderPipeline(DiffusionPipeline):
58+
class StableCascadeDecoderPipeline(DeprecatedPipelineMixin, DiffusionPipeline):
5959
"""
6060
Pipeline for generating images from the Stable Cascade model.
6161
@@ -79,6 +79,8 @@ class StableCascadeDecoderPipeline(DiffusionPipeline):
7979
width=int(24*10.67)=256 in order to match the training conditions.
8080
"""
8181

82+
_last_supported_version = "0.35.2"
83+
8284
unet_name = "decoder"
8385
text_encoder_name = "text_encoder"
8486
model_cpu_offload_seq = "text_encoder->decoder->vqgan"

src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_combined.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
from ...models import StableCascadeUNet
2121
from ...schedulers import DDPMWuerstchenScheduler
2222
from ...utils import is_torch_version, replace_example_docstring
23-
from ..pipeline_utils import DiffusionPipeline
23+
from ..pipeline_utils import DeprecatedPipelineMixin, DiffusionPipeline
2424
from ..wuerstchen.modeling_paella_vq_model import PaellaVQModel
2525
from .pipeline_stable_cascade import StableCascadeDecoderPipeline
2626
from .pipeline_stable_cascade_prior import StableCascadePriorPipeline
@@ -42,7 +42,7 @@
4242
"""
4343

4444

45-
class StableCascadeCombinedPipeline(DiffusionPipeline):
45+
class StableCascadeCombinedPipeline(DeprecatedPipelineMixin, DiffusionPipeline):
4646
"""
4747
Combined Pipeline for text-to-image generation using Stable Cascade.
4848
@@ -74,6 +74,8 @@ class StableCascadeCombinedPipeline(DiffusionPipeline):
7474
Frozen CLIP image-encoder ([clip-vit-large-patch14](https://huggingface.co/openai/clip-vit-large-patch14)).
7575
"""
7676

77+
_last_supported_version = "0.35.2"
78+
7779
_load_connected_pipes = True
7880
_optional_components = ["prior_feature_extractor", "prior_image_encoder"]
7981

src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_prior.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
from ...schedulers import DDPMWuerstchenScheduler
2626
from ...utils import BaseOutput, is_torch_xla_available, logging, replace_example_docstring
2727
from ...utils.torch_utils import randn_tensor
28-
from ..pipeline_utils import DiffusionPipeline
28+
from ..pipeline_utils import DeprecatedPipelineMixin, DiffusionPipeline
2929

3030

3131
if is_torch_xla_available():
@@ -77,7 +77,7 @@ class StableCascadePriorPipelineOutput(BaseOutput):
7777
negative_prompt_embeds_pooled: Union[torch.Tensor, np.ndarray]
7878

7979

80-
class StableCascadePriorPipeline(DiffusionPipeline):
80+
class StableCascadePriorPipeline(DeprecatedPipelineMixin, DiffusionPipeline):
8181
"""
8282
Pipeline for generating image prior for Stable Cascade.
8383
@@ -103,6 +103,8 @@ class StableCascadePriorPipeline(DiffusionPipeline):
103103
Default resolution for multiple images generated.
104104
"""
105105

106+
_last_supported_version = "0.35.2"
107+
106108
unet_name = "prior"
107109
text_encoder_name = "text_encoder"
108110
model_cpu_offload_seq = "image_encoder->text_encoder->prior"

0 commit comments

Comments
 (0)