Skip to content

Commit e5480d8

Browse files
committed
fix?
1 parent 34c60b1 commit e5480d8

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",
@@ -840,6 +840,12 @@
840840
apply_layer_skip,
841841
apply_pyramid_attention_broadcast,
842842
)
843+
from .image_processor import (
844+
IPAdapterMaskProcessor,
845+
PixArtImageProcessor,
846+
VaeImageProcessor,
847+
VaeImageProcessorLDM3D,
848+
)
843849
from .models import (
844850
AllegroTransformer3DModel,
845851
AsymmetricAutoencoderKL,
@@ -958,13 +964,6 @@
958964
ScoreSdeVePipeline,
959965
StableDiffusionMixin,
960966
)
961-
from .image_processor import (
962-
VaeImageProcessor,
963-
VaeImageProcessorLDM3D,
964-
PixArtImageProcessor,
965-
IPAdapterMaskProcessor,
966-
)
967-
from .video_processor import VideoProcessor
968967
from .quantizers import DiffusersQuantizer
969968
from .schedulers import (
970969
AmusedScheduler,
@@ -1006,6 +1005,7 @@
10061005
VQDiffusionScheduler,
10071006
)
10081007
from .training_utils import EMAModel
1008+
from .video_processor import VideoProcessor
10091009

10101010
try:
10111011
if not (is_torch_available() and is_scipy_available()):

0 commit comments

Comments
 (0)