Skip to content

Commit a6505c2

Browse files
committed
deprecate blip.
1 parent 2e83cbb commit a6505c2

File tree

10 files changed

+66
-8
lines changed

10 files changed

+66
-8
lines changed

src/diffusers/pipelines/__init__.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,6 @@
146146
"AudioLDM2ProjectionModel",
147147
"AudioLDM2UNet2DConditionModel",
148148
]
149-
_import_structure["blip_diffusion"] = ["BlipDiffusionPipeline"]
150149
_import_structure["cogvideo"] = [
151150
"CogVideoXPipeline",
152151
"CogVideoXImageToVideoPipeline",
@@ -158,7 +157,6 @@
158157
_import_structure["consisid"] = ["ConsisIDPipeline"]
159158
_import_structure["controlnet"].extend(
160159
[
161-
"BlipDiffusionControlNetPipeline",
162160
"StableDiffusionControlNetImg2ImgPipeline",
163161
"StableDiffusionControlNetInpaintPipeline",
164162
"StableDiffusionControlNetPipeline",
@@ -503,7 +501,6 @@
503501
AudioLDM2UNet2DConditionModel,
504502
)
505503
from .aura_flow import AuraFlowPipeline
506-
from .blip_diffusion import BlipDiffusionPipeline
507504
from .cogvideo import (
508505
CogVideoXFunControlPipeline,
509506
CogVideoXImageToVideoPipeline,
@@ -514,7 +511,6 @@
514511
from .cogview4 import CogView4ControlPipeline, CogView4Pipeline
515512
from .consisid import ConsisIDPipeline
516513
from .controlnet import (
517-
BlipDiffusionControlNetPipeline,
518514
StableDiffusionControlNetImg2ImgPipeline,
519515
StableDiffusionControlNetInpaintPipeline,
520516
StableDiffusionControlNetPipeline,
@@ -544,6 +540,8 @@
544540
from .deprecated import (
545541
AltDiffusionImg2ImgPipeline,
546542
AltDiffusionPipeline,
543+
BlipDiffusionControlNetPipeline,
544+
BlipDiffusionPipeline,
547545
CycleDiffusionPipeline,
548546
StableDiffusionInpaintPipelineLegacy,
549547
StableDiffusionModelEditingPipeline,

src/diffusers/pipelines/controlnet/__init__.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@
2424
else:
2525
_import_structure["multicontrolnet"] = ["MultiControlNetModel"]
2626
_import_structure["pipeline_controlnet"] = ["StableDiffusionControlNetPipeline"]
27-
_import_structure["pipeline_controlnet_blip_diffusion"] = ["BlipDiffusionControlNetPipeline"]
2827
_import_structure["pipeline_controlnet_img2img"] = ["StableDiffusionControlNetImg2ImgPipeline"]
2928
_import_structure["pipeline_controlnet_inpaint"] = ["StableDiffusionControlNetInpaintPipeline"]
3029
_import_structure["pipeline_controlnet_inpaint_sd_xl"] = ["StableDiffusionXLControlNetInpaintPipeline"]
@@ -54,7 +53,6 @@
5453
else:
5554
from .multicontrolnet import MultiControlNetModel
5655
from .pipeline_controlnet import StableDiffusionControlNetPipeline
57-
from .pipeline_controlnet_blip_diffusion import BlipDiffusionControlNetPipeline
5856
from .pipeline_controlnet_img2img import StableDiffusionControlNetImg2ImgPipeline
5957
from .pipeline_controlnet_inpaint import StableDiffusionControlNetInpaintPipeline
6058
from .pipeline_controlnet_inpaint_sd_xl import StableDiffusionXLControlNetInpaintPipeline

src/diffusers/pipelines/deprecated/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@
4242
"AltDiffusionPipeline",
4343
"AltDiffusionPipelineOutput",
4444
]
45+
_import_structure["blip_diffusion"] = ["BlipDiffusionPipeline"]
4546
_import_structure["versatile_diffusion"] = [
4647
"VersatileDiffusionDualGuidedPipeline",
4748
"VersatileDiffusionImageVariationPipeline",
@@ -56,6 +57,7 @@
5657
"StableDiffusionParadigmsPipeline",
5758
"StableDiffusionModelEditingPipeline",
5859
]
60+
_import_structure["controlnet"] = ["BlipDiffusionControlNetPipeline"]
5961

6062
try:
6163
if not (is_torch_available() and is_librosa_available()):
@@ -103,6 +105,8 @@
103105
else:
104106
from .alt_diffusion import AltDiffusionImg2ImgPipeline, AltDiffusionPipeline, AltDiffusionPipelineOutput
105107
from .audio_diffusion import AudioDiffusionPipeline, Mel
108+
from .blip_diffusion import BlipDiffusionPipeline
109+
from .controlnet import BlipDiffusionControlNetPipeline
106110
from .spectrogram_diffusion import SpectrogramDiffusionPipeline
107111
from .stable_diffusion_variants import (
108112
CycleDiffusionPipeline,

src/diffusers/pipelines/blip_diffusion/__init__.py renamed to src/diffusers/pipelines/deprecated/blip_diffusion/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,14 +5,14 @@
55
import PIL
66
from PIL import Image
77

8-
from ...utils import OptionalDependencyNotAvailable, is_torch_available, is_transformers_available
8+
from ....utils import OptionalDependencyNotAvailable, is_torch_available, is_transformers_available
99

1010

1111
try:
1212
if not (is_transformers_available() and is_torch_available()):
1313
raise OptionalDependencyNotAvailable()
1414
except OptionalDependencyNotAvailable:
15-
from ...utils.dummy_torch_and_transformers_objects import ShapEPipeline
15+
from ....utils.dummy_torch_and_transformers_objects import ShapEPipeline
1616
else:
1717
from .blip_image_processing import BlipImageProcessor
1818
from .modeling_blip2 import Blip2QFormerModel
Lines changed: 58 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,58 @@
1+
from typing import TYPE_CHECKING
2+
3+
from ....utils import (
4+
DIFFUSERS_SLOW_IMPORT,
5+
OptionalDependencyNotAvailable,
6+
_LazyModule,
7+
get_objects_from_module,
8+
is_flax_available,
9+
is_torch_available,
10+
is_transformers_available,
11+
)
12+
13+
14+
_dummy_objects = {}
15+
_import_structure = {}
16+
17+
try:
18+
if not (is_transformers_available() and is_torch_available()):
19+
raise OptionalDependencyNotAvailable()
20+
except OptionalDependencyNotAvailable:
21+
from ....utils import dummy_torch_and_transformers_objects # noqa F403
22+
23+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
24+
else:
25+
_import_structure["pipeline_controlnet_blip_diffusion"] = ["BlipDiffusionControlNetPipeline"]
26+
try:
27+
if not (is_transformers_available() and is_flax_available()):
28+
raise OptionalDependencyNotAvailable()
29+
except OptionalDependencyNotAvailable:
30+
from ...utils import dummy_flax_and_transformers_objects # noqa F403
31+
32+
_dummy_objects.update(get_objects_from_module(dummy_flax_and_transformers_objects))
33+
else:
34+
_import_structure["pipeline_flax_controlnet"] = ["FlaxStableDiffusionControlNetPipeline"]
35+
36+
37+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
38+
try:
39+
if not (is_transformers_available() and is_torch_available()):
40+
raise OptionalDependencyNotAvailable()
41+
42+
except OptionalDependencyNotAvailable:
43+
from ....utils.dummy_torch_and_transformers_objects import *
44+
else:
45+
from .pipeline_controlnet_blip_diffusion import BlipDiffusionControlNetPipeline
46+
47+
48+
else:
49+
import sys
50+
51+
sys.modules[__name__] = _LazyModule(
52+
__name__,
53+
globals()["__file__"],
54+
_import_structure,
55+
module_spec=__spec__,
56+
)
57+
for name, value in _dummy_objects.items():
58+
setattr(sys.modules[__name__], name, value)

0 commit comments

Comments
 (0)