Unverified Commit 2c302879 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[chore] deprecate blip controlnet pipeline. (#11877)

* deprecate blip controlnet pipeline.

* last_supported_version
parent 425a715e
...@@ -29,7 +29,7 @@ from ...utils.torch_utils import randn_tensor ...@@ -29,7 +29,7 @@ from ...utils.torch_utils import randn_tensor
from ..blip_diffusion.blip_image_processing import BlipImageProcessor from ..blip_diffusion.blip_image_processing import BlipImageProcessor
from ..blip_diffusion.modeling_blip2 import Blip2QFormerModel from ..blip_diffusion.modeling_blip2 import Blip2QFormerModel
from ..blip_diffusion.modeling_ctx_clip import ContextCLIPTextModel from ..blip_diffusion.modeling_ctx_clip import ContextCLIPTextModel
from ..pipeline_utils import DiffusionPipeline, ImagePipelineOutput from ..pipeline_utils import DeprecatedPipelineMixin, DiffusionPipeline, ImagePipelineOutput
if is_torch_xla_available(): if is_torch_xla_available():
...@@ -88,7 +88,7 @@ EXAMPLE_DOC_STRING = """ ...@@ -88,7 +88,7 @@ EXAMPLE_DOC_STRING = """
""" """
class BlipDiffusionControlNetPipeline(DiffusionPipeline): class BlipDiffusionControlNetPipeline(DeprecatedPipelineMixin, DiffusionPipeline):
""" """
Pipeline for Canny Edge based Controlled subject-driven generation using Blip Diffusion. Pipeline for Canny Edge based Controlled subject-driven generation using Blip Diffusion.
...@@ -116,6 +116,7 @@ class BlipDiffusionControlNetPipeline(DiffusionPipeline): ...@@ -116,6 +116,7 @@ class BlipDiffusionControlNetPipeline(DiffusionPipeline):
Position of the context token in the text encoder. Position of the context token in the text encoder.
""" """
_last_supported_version = "0.33.1"
model_cpu_offload_seq = "qformer->text_encoder->unet->vae" model_cpu_offload_seq = "qformer->text_encoder->unet->vae"
def __init__( def __init__(
......
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