Skip to content

Commit 7e6743a

Browse files
committed
fix?
1 parent 1a36051 commit 7e6743a

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

src/diffusers/__init__.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -36,12 +36,6 @@
3636
"configuration_utils": ["ConfigMixin"],
3737
"guiders": [],
3838
"hooks": [],
39-
"image_processor": [
40-
"VaeImageProcessor",
41-
"VaeImageProcessorLDM3D",
42-
"PixArtImageProcessor",
43-
"IPAdapterMaskProcessor",
44-
],
4539
"video_processor": [
4640
"VideoProcessor",
4741
],
@@ -169,6 +163,12 @@
169163
"apply_pyramid_attention_broadcast",
170164
]
171165
)
166+
_import_structure["image_processor"] = [
167+
"IPAdapterMaskProcessor",
168+
"PixArtImageProcessor",
169+
"VaeImageProcessor",
170+
"VaeImageProcessorLDM3D",
171+
]
172172
_import_structure["models"].extend(
173173
[
174174
"AllegroTransformer3DModel",
@@ -834,6 +834,12 @@
834834
apply_layer_skip,
835835
apply_pyramid_attention_broadcast,
836836
)
837+
from .image_processor import (
838+
IPAdapterMaskProcessor,
839+
PixArtImageProcessor,
840+
VaeImageProcessor,
841+
VaeImageProcessorLDM3D,
842+
)
837843
from .models import (
838844
AllegroTransformer3DModel,
839845
AsymmetricAutoencoderKL,
@@ -949,13 +955,6 @@
949955
ScoreSdeVePipeline,
950956
StableDiffusionMixin,
951957
)
952-
from .image_processor import (
953-
VaeImageProcessor,
954-
VaeImageProcessorLDM3D,
955-
PixArtImageProcessor,
956-
IPAdapterMaskProcessor,
957-
)
958-
from .video_processor import VideoProcessor
959958
from .quantizers import DiffusersQuantizer
960959
from .schedulers import (
961960
AmusedScheduler,
@@ -997,6 +996,7 @@
997996
VQDiffusionScheduler,
998997
)
999998
from .training_utils import EMAModel
999+
from .video_processor import VideoProcessor
10001000

10011001
try:
10021002
if not (is_torch_available() and is_scipy_available()):

0 commit comments

Comments
 (0)