__init__.py 43.7 KB
Newer Older
1
__version__ = "0.33.0.dev0"
2

Dhruv Nair's avatar
Dhruv Nair committed
3
4
from typing import TYPE_CHECKING

5
from .utils import (
6
    DIFFUSERS_SLOW_IMPORT,
7
    OptionalDependencyNotAvailable,
Dhruv Nair's avatar
Dhruv Nair committed
8
    _LazyModule,
9
10
    is_accelerate_available,
    is_bitsandbytes_available,
11
    is_flax_available,
12
    is_gguf_available,
13
    is_k_diffusion_available,
14
    is_librosa_available,
15
    is_note_seq_available,
16
    is_onnx_available,
Sayak Paul's avatar
Sayak Paul committed
17
    is_opencv_available,
18
    is_optimum_quanto_available,
19
    is_scipy_available,
20
    is_sentencepiece_available,
21
    is_torch_available,
22
    is_torchao_available,
23
    is_torchsde_available,
24
25
    is_transformers_available,
)
Patrick von Platen's avatar
Patrick von Platen committed
26
27


Dhruv Nair's avatar
Dhruv Nair committed
28
29
30
31
32
33
34
35
36
# Lazy Import based on
# https://github.com/huggingface/transformers/blob/main/src/transformers/__init__.py

# When adding a new object to this init, please add it to `_import_structure`. The `_import_structure` is a dictionary submodule to list of object names,
# and is used to defer the actual importing for when the objects are requested.
# This way `import diffusers` provides the names in the namespace without actually importing anything (and especially none of the backends).

_import_structure = {
    "configuration_utils": ["ConfigMixin"],
37
    "hooks": [],
38
    "loaders": ["FromOriginalModelMixin"],
Dhruv Nair's avatar
Dhruv Nair committed
39
40
    "models": [],
    "pipelines": [],
41
    "quantizers.quantization_config": [],
Dhruv Nair's avatar
Dhruv Nair committed
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
    "schedulers": [],
    "utils": [
        "OptionalDependencyNotAvailable",
        "is_flax_available",
        "is_inflect_available",
        "is_invisible_watermark_available",
        "is_k_diffusion_available",
        "is_k_diffusion_version",
        "is_librosa_available",
        "is_note_seq_available",
        "is_onnx_available",
        "is_scipy_available",
        "is_torch_available",
        "is_torchsde_available",
        "is_transformers_available",
        "is_transformers_version",
        "is_unidecode_available",
        "logging",
    ],
}

63
try:
64
    if not is_torch_available() and not is_accelerate_available() and not is_bitsandbytes_available():
65
66
67
68
69
70
71
72
73
74
75
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    from .utils import dummy_bitsandbytes_objects

    _import_structure["utils.dummy_bitsandbytes_objects"] = [
        name for name in dir(dummy_bitsandbytes_objects) if not name.startswith("_")
    ]
else:
    _import_structure["quantizers.quantization_config"].append("BitsAndBytesConfig")

try:
76
    if not is_torch_available() and not is_accelerate_available() and not is_gguf_available():
77
78
79
80
81
82
83
84
85
86
87
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    from .utils import dummy_gguf_objects

    _import_structure["utils.dummy_gguf_objects"] = [
        name for name in dir(dummy_gguf_objects) if not name.startswith("_")
    ]
else:
    _import_structure["quantizers.quantization_config"].append("GGUFQuantizationConfig")

try:
88
    if not is_torch_available() and not is_accelerate_available() and not is_torchao_available():
89
90
91
92
93
94
95
96
97
98
99
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    from .utils import dummy_torchao_objects

    _import_structure["utils.dummy_torchao_objects"] = [
        name for name in dir(dummy_torchao_objects) if not name.startswith("_")
    ]
else:
    _import_structure["quantizers.quantization_config"].append("TorchAoConfig")

try:
100
    if not is_torch_available() and not is_accelerate_available() and not is_optimum_quanto_available():
101
102
103
104
105
106
107
108
109
110
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    from .utils import dummy_optimum_quanto_objects

    _import_structure["utils.dummy_optimum_quanto_objects"] = [
        name for name in dir(dummy_optimum_quanto_objects) if not name.startswith("_")
    ]
else:
    _import_structure["quantizers.quantization_config"].append("QuantoConfig")

111
112
113
114
try:
    if not is_onnx_available():
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
115
116
117
118
119
120
    from .utils import dummy_onnx_objects  # noqa F403

    _import_structure["utils.dummy_onnx_objects"] = [
        name for name in dir(dummy_onnx_objects) if not name.startswith("_")
    ]

121
else:
Dhruv Nair's avatar
Dhruv Nair committed
122
    _import_structure["pipelines"].extend(["OnnxRuntimeModel"])
123

124
125
126
127
try:
    if not is_torch_available():
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
128
129
130
131
    from .utils import dummy_pt_objects  # noqa F403

    _import_structure["utils.dummy_pt_objects"] = [name for name in dir(dummy_pt_objects) if not name.startswith("_")]

132
else:
133
134
    _import_structure["hooks"].extend(
        [
Aryan's avatar
Aryan committed
135
            "FasterCacheConfig",
136
137
            "HookRegistry",
            "PyramidAttentionBroadcastConfig",
Aryan's avatar
Aryan committed
138
            "apply_faster_cache",
139
140
141
            "apply_pyramid_attention_broadcast",
        ]
    )
Dhruv Nair's avatar
Dhruv Nair committed
142
143
    _import_structure["models"].extend(
        [
Aryan's avatar
Aryan committed
144
            "AllegroTransformer3DModel",
Dhruv Nair's avatar
Dhruv Nair committed
145
            "AsymmetricAutoencoderKL",
Sayak Paul's avatar
Sayak Paul committed
146
            "AuraFlowTransformer2DModel",
147
            "AutoencoderDC",
Dhruv Nair's avatar
Dhruv Nair committed
148
            "AutoencoderKL",
Aryan's avatar
Aryan committed
149
            "AutoencoderKLAllegro",
150
            "AutoencoderKLCogVideoX",
Aryan's avatar
Aryan committed
151
            "AutoencoderKLHunyuanVideo",
Aryan's avatar
Aryan committed
152
            "AutoencoderKLLTXVideo",
153
            "AutoencoderKLMagvit",
Aryan's avatar
Aryan committed
154
            "AutoencoderKLMochi",
Suraj Patil's avatar
Suraj Patil committed
155
            "AutoencoderKLTemporalDecoder",
156
            "AutoencoderKLWan",
157
            "AutoencoderOobleck",
Dhruv Nair's avatar
Dhruv Nair committed
158
            "AutoencoderTiny",
hlky's avatar
hlky committed
159
            "AutoModel",
160
            "CacheMixin",
161
            "CogVideoXTransformer3DModel",
Yuxuan.Zhang's avatar
Yuxuan.Zhang committed
162
            "CogView3PlusTransformer2DModel",
163
            "CogView4Transformer2DModel",
Shenghai Yuan's avatar
Shenghai Yuan committed
164
            "ConsisIDTransformer3DModel",
Will Berman's avatar
Will Berman committed
165
            "ConsistencyDecoderVAE",
Dhruv Nair's avatar
Dhruv Nair committed
166
            "ControlNetModel",
hlky's avatar
hlky committed
167
            "ControlNetUnionModel",
168
            "ControlNetXSAdapter",
169
            "DiTTransformer2DModel",
170
            "EasyAnimateTransformer3DModel",
王奇勋's avatar
王奇勋 committed
171
            "FluxControlNetModel",
172
            "FluxMultiControlNetModel",
Sayak Paul's avatar
Sayak Paul committed
173
            "FluxTransformer2DModel",
174
            "HunyuanDiT2DControlNetModel",
175
            "HunyuanDiT2DModel",
176
            "HunyuanDiT2DMultiControlNetModel",
Aryan's avatar
Aryan committed
177
            "HunyuanVideoTransformer3DModel",
178
            "I2VGenXLUNet",
179
            "Kandinsky3UNet",
180
            "LatteTransformer3DModel",
Aryan's avatar
Aryan committed
181
            "LTXVideoTransformer3DModel",
Le Zhuo's avatar
Le Zhuo committed
182
            "Lumina2Transformer2DModel",
183
            "LuminaNextDiT2DModel",
Aryan's avatar
Aryan committed
184
            "MochiTransformer3DModel",
Dhruv Nair's avatar
Dhruv Nair committed
185
            "ModelMixin",
Dhruv Nair's avatar
Dhruv Nair committed
186
            "MotionAdapter",
Dhruv Nair's avatar
Dhruv Nair committed
187
            "MultiAdapter",
188
            "MultiControlNetModel",
Shitao Xiao's avatar
Shitao Xiao committed
189
            "OmniGenTransformer2DModel",
190
            "PixArtTransformer2DModel",
Dhruv Nair's avatar
Dhruv Nair committed
191
            "PriorTransformer",
192
            "SanaTransformer2DModel",
193
194
            "SD3ControlNetModel",
            "SD3MultiControlNetModel",
Dhruv Nair's avatar
Dhruv Nair committed
195
            "SD3Transformer2DModel",
Aryan's avatar
Aryan committed
196
            "SparseControlNetModel",
197
            "StableAudioDiTModel",
198
            "StableCascadeUNet",
Dhruv Nair's avatar
Dhruv Nair committed
199
200
201
            "T2IAdapter",
            "T5FilmDecoder",
            "Transformer2DModel",
hlky's avatar
hlky committed
202
            "TransformerTemporalModel",
Dhruv Nair's avatar
Dhruv Nair committed
203
204
205
206
            "UNet1DModel",
            "UNet2DConditionModel",
            "UNet2DModel",
            "UNet3DConditionModel",
207
            "UNetControlNetXSModel",
Dhruv Nair's avatar
Dhruv Nair committed
208
            "UNetMotionModel",
Suraj Patil's avatar
Suraj Patil committed
209
            "UNetSpatioTemporalConditionModel",
Will Berman's avatar
Will Berman committed
210
            "UVit2DModel",
Dhruv Nair's avatar
Dhruv Nair committed
211
            "VQModel",
212
            "WanTransformer3DModel",
Dhruv Nair's avatar
Dhruv Nair committed
213
        ]
214
    )
Dhruv Nair's avatar
Dhruv Nair committed
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
    _import_structure["optimization"] = [
        "get_constant_schedule",
        "get_constant_schedule_with_warmup",
        "get_cosine_schedule_with_warmup",
        "get_cosine_with_hard_restarts_schedule_with_warmup",
        "get_linear_schedule_with_warmup",
        "get_polynomial_decay_schedule_with_warmup",
        "get_scheduler",
    ]
    _import_structure["pipelines"].extend(
        [
            "AudioPipelineOutput",
            "AutoPipelineForImage2Image",
            "AutoPipelineForInpainting",
            "AutoPipelineForText2Image",
            "ConsistencyModelPipeline",
            "DanceDiffusionPipeline",
            "DDIMPipeline",
            "DDPMPipeline",
            "DiffusionPipeline",
            "DiTPipeline",
            "ImagePipelineOutput",
            "KarrasVePipeline",
            "LDMPipeline",
            "LDMSuperResolutionPipeline",
            "PNDMPipeline",
            "RePaintPipeline",
            "ScoreSdeVePipeline",
243
            "StableDiffusionMixin",
Dhruv Nair's avatar
Dhruv Nair committed
244
        ]
245
    )
246
    _import_structure["quantizers"] = ["DiffusersQuantizer"]
Dhruv Nair's avatar
Dhruv Nair committed
247
248
    _import_structure["schedulers"].extend(
        [
Will Berman's avatar
Will Berman committed
249
            "AmusedScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
250
            "CMStochasticIterativeScheduler",
251
252
            "CogVideoXDDIMScheduler",
            "CogVideoXDPMScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
253
254
255
256
257
258
259
260
261
262
            "DDIMInverseScheduler",
            "DDIMParallelScheduler",
            "DDIMScheduler",
            "DDPMParallelScheduler",
            "DDPMScheduler",
            "DDPMWuerstchenScheduler",
            "DEISMultistepScheduler",
            "DPMSolverMultistepInverseScheduler",
            "DPMSolverMultistepScheduler",
            "DPMSolverSinglestepScheduler",
263
            "EDMDPMSolverMultistepScheduler",
Suraj Patil's avatar
Suraj Patil committed
264
            "EDMEulerScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
265
266
            "EulerAncestralDiscreteScheduler",
            "EulerDiscreteScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
267
            "FlowMatchEulerDiscreteScheduler",
268
            "FlowMatchHeunDiscreteScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
269
270
271
272
273
            "HeunDiscreteScheduler",
            "IPNDMScheduler",
            "KarrasVeScheduler",
            "KDPM2AncestralDiscreteScheduler",
            "KDPM2DiscreteScheduler",
274
            "LCMScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
275
276
            "PNDMScheduler",
            "RePaintScheduler",
Junsong Chen's avatar
Junsong Chen committed
277
            "SASolverScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
278
            "SchedulerMixin",
YiYi Xu's avatar
YiYi Xu committed
279
            "SCMScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
280
            "ScoreSdeVeScheduler",
Michael's avatar
Michael committed
281
            "TCDScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
282
283
284
285
            "UnCLIPScheduler",
            "UniPCMultistepScheduler",
            "VQDiffusionScheduler",
        ]
286
    )
Dhruv Nair's avatar
Dhruv Nair committed
287
    _import_structure["training_utils"] = ["EMAModel"]
Patrick von Platen's avatar
Patrick von Platen committed
288

289
290
291
292
try:
    if not (is_torch_available() and is_scipy_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
293
294
295
296
297
298
    from .utils import dummy_torch_and_scipy_objects  # noqa F403

    _import_structure["utils.dummy_torch_and_scipy_objects"] = [
        name for name in dir(dummy_torch_and_scipy_objects) if not name.startswith("_")
    ]

299
else:
Dhruv Nair's avatar
Dhruv Nair committed
300
    _import_structure["schedulers"].extend(["LMSDiscreteScheduler"])
Patrick von Platen's avatar
Patrick von Platen committed
301

302
303
304
305
try:
    if not (is_torch_available() and is_torchsde_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
306
307
308
309
310
311
    from .utils import dummy_torch_and_torchsde_objects  # noqa F403

    _import_structure["utils.dummy_torch_and_torchsde_objects"] = [
        name for name in dir(dummy_torch_and_torchsde_objects) if not name.startswith("_")
    ]

312
else:
313
    _import_structure["schedulers"].extend(["CosineDPMSolverMultistepScheduler", "DPMSolverSDEScheduler"])
314
315
316
317
318

try:
    if not (is_torch_available() and is_transformers_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
319
320
321
322
323
324
    from .utils import dummy_torch_and_transformers_objects  # noqa F403

    _import_structure["utils.dummy_torch_and_transformers_objects"] = [
        name for name in dir(dummy_torch_and_transformers_objects) if not name.startswith("_")
    ]

325
else:
Dhruv Nair's avatar
Dhruv Nair committed
326
327
    _import_structure["pipelines"].extend(
        [
Aryan's avatar
Aryan committed
328
            "AllegroPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
329
330
            "AltDiffusionImg2ImgPipeline",
            "AltDiffusionPipeline",
Will Berman's avatar
Will Berman committed
331
332
333
            "AmusedImg2ImgPipeline",
            "AmusedInpaintPipeline",
            "AmusedPipeline",
334
            "AnimateDiffControlNetPipeline",
Aryan's avatar
Aryan committed
335
            "AnimateDiffPAGPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
336
            "AnimateDiffPipeline",
Aryan's avatar
Aryan committed
337
            "AnimateDiffSDXLPipeline",
Aryan's avatar
Aryan committed
338
            "AnimateDiffSparseControlNetPipeline",
339
            "AnimateDiffVideoToVideoControlNetPipeline",
Aryan V S's avatar
Aryan V S committed
340
            "AnimateDiffVideoToVideoPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
341
342
343
344
            "AudioLDM2Pipeline",
            "AudioLDM2ProjectionModel",
            "AudioLDM2UNet2DConditionModel",
            "AudioLDMPipeline",
Sayak Paul's avatar
Sayak Paul committed
345
            "AuraFlowPipeline",
Ayush Mangal's avatar
Ayush Mangal committed
346
347
            "BlipDiffusionControlNetPipeline",
            "BlipDiffusionPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
348
            "CLIPImageProjection",
349
            "CogVideoXFunControlPipeline",
350
            "CogVideoXImageToVideoPipeline",
351
            "CogVideoXPipeline",
352
            "CogVideoXVideoToVideoPipeline",
Yuxuan.Zhang's avatar
Yuxuan.Zhang committed
353
            "CogView3PlusPipeline",
Yuxuan Zhang's avatar
Yuxuan Zhang committed
354
            "CogView4ControlPipeline",
355
            "CogView4Pipeline",
Dhruv Nair's avatar
Dhruv Nair committed
356
            "CycleDiffusionPipeline",
357
358
359
            "EasyAnimateControlPipeline",
            "EasyAnimateInpaintPipeline",
            "EasyAnimatePipeline",
Aryan's avatar
Aryan committed
360
            "FluxControlImg2ImgPipeline",
361
            "FluxControlInpaintPipeline",
362
363
            "FluxControlNetImg2ImgPipeline",
            "FluxControlNetInpaintPipeline",
王奇勋's avatar
王奇勋 committed
364
            "FluxControlNetPipeline",
Aryan's avatar
Aryan committed
365
366
            "FluxControlPipeline",
            "FluxFillPipeline",
367
368
            "FluxImg2ImgPipeline",
            "FluxInpaintPipeline",
Sayak Paul's avatar
Sayak Paul committed
369
            "FluxPipeline",
Aryan's avatar
Aryan committed
370
            "FluxPriorReduxPipeline",
371
            "HunyuanDiTControlNetPipeline",
372
            "HunyuanDiTPAGPipeline",
373
            "HunyuanDiTPipeline",
Aryan's avatar
Aryan committed
374
            "HunyuanSkyreelsImageToVideoPipeline",
Aryan's avatar
Aryan committed
375
            "HunyuanVideoImageToVideoPipeline",
Aryan's avatar
Aryan committed
376
            "HunyuanVideoPipeline",
377
            "I2VGenXLPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
378
379
380
381
382
383
384
            "IFImg2ImgPipeline",
            "IFImg2ImgSuperResolutionPipeline",
            "IFInpaintingPipeline",
            "IFInpaintingSuperResolutionPipeline",
            "IFPipeline",
            "IFSuperResolutionPipeline",
            "ImageTextPipelineOutput",
385
386
            "Kandinsky3Img2ImgPipeline",
            "Kandinsky3Pipeline",
Dhruv Nair's avatar
Dhruv Nair committed
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
            "KandinskyCombinedPipeline",
            "KandinskyImg2ImgCombinedPipeline",
            "KandinskyImg2ImgPipeline",
            "KandinskyInpaintCombinedPipeline",
            "KandinskyInpaintPipeline",
            "KandinskyPipeline",
            "KandinskyPriorPipeline",
            "KandinskyV22CombinedPipeline",
            "KandinskyV22ControlnetImg2ImgPipeline",
            "KandinskyV22ControlnetPipeline",
            "KandinskyV22Img2ImgCombinedPipeline",
            "KandinskyV22Img2ImgPipeline",
            "KandinskyV22InpaintCombinedPipeline",
            "KandinskyV22InpaintPipeline",
            "KandinskyV22Pipeline",
            "KandinskyV22PriorEmb2EmbPipeline",
            "KandinskyV22PriorPipeline",
404
            "LatentConsistencyModelImg2ImgPipeline",
405
            "LatentConsistencyModelPipeline",
406
            "LattePipeline",
Dhruv Nair's avatar
Dhruv Nair committed
407
            "LDMTextToImagePipeline",
408
409
            "LEditsPPPipelineStableDiffusion",
            "LEditsPPPipelineStableDiffusionXL",
Aryan's avatar
Aryan committed
410
            "LTXConditionPipeline",
Aryan's avatar
Aryan committed
411
412
            "LTXImageToVideoPipeline",
            "LTXPipeline",
413
            "Lumina2Pipeline",
Le Zhuo's avatar
Le Zhuo committed
414
            "Lumina2Text2ImgPipeline",
415
            "LuminaPipeline",
416
            "LuminaText2ImgPipeline",
417
            "MarigoldDepthPipeline",
418
            "MarigoldIntrinsicsPipeline",
419
            "MarigoldNormalsPipeline",
Aryan's avatar
Aryan committed
420
            "MochiPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
421
            "MusicLDMPipeline",
Shitao Xiao's avatar
Shitao Xiao committed
422
            "OmniGenPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
423
            "PaintByExamplePipeline",
Dhruv Nair's avatar
Dhruv Nair committed
424
            "PIAPipeline",
Sayak Paul's avatar
Sayak Paul committed
425
            "PixArtAlphaPipeline",
426
            "PixArtSigmaPAGPipeline",
427
            "PixArtSigmaPipeline",
Aryan's avatar
Aryan committed
428
            "ReduxImageEncoder",
429
430
            "SanaPAGPipeline",
            "SanaPipeline",
YiYi Xu's avatar
YiYi Xu committed
431
            "SanaSprintPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
432
433
434
            "SemanticStableDiffusionPipeline",
            "ShapEImg2ImgPipeline",
            "ShapEPipeline",
435
436
            "StableAudioPipeline",
            "StableAudioProjectionModel",
437
438
439
            "StableCascadeCombinedPipeline",
            "StableCascadeDecoderPipeline",
            "StableCascadePriorPipeline",
440
            "StableDiffusion3ControlNetInpaintingPipeline",
441
            "StableDiffusion3ControlNetPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
442
            "StableDiffusion3Img2ImgPipeline",
443
            "StableDiffusion3InpaintPipeline",
444
            "StableDiffusion3PAGImg2ImgPipeline",
445
            "StableDiffusion3PAGImg2ImgPipeline",
StAlKeR7779's avatar
StAlKeR7779 committed
446
            "StableDiffusion3PAGPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
447
            "StableDiffusion3Pipeline",
Dhruv Nair's avatar
Dhruv Nair committed
448
449
450
451
            "StableDiffusionAdapterPipeline",
            "StableDiffusionAttendAndExcitePipeline",
            "StableDiffusionControlNetImg2ImgPipeline",
            "StableDiffusionControlNetInpaintPipeline",
452
            "StableDiffusionControlNetPAGInpaintPipeline",
453
            "StableDiffusionControlNetPAGPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
454
            "StableDiffusionControlNetPipeline",
455
            "StableDiffusionControlNetXSPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
456
457
458
459
460
461
462
463
464
465
466
467
            "StableDiffusionDepth2ImgPipeline",
            "StableDiffusionDiffEditPipeline",
            "StableDiffusionGLIGENPipeline",
            "StableDiffusionGLIGENTextImagePipeline",
            "StableDiffusionImageVariationPipeline",
            "StableDiffusionImg2ImgPipeline",
            "StableDiffusionInpaintPipeline",
            "StableDiffusionInpaintPipelineLegacy",
            "StableDiffusionInstructPix2PixPipeline",
            "StableDiffusionLatentUpscalePipeline",
            "StableDiffusionLDM3DPipeline",
            "StableDiffusionModelEditingPipeline",
468
            "StableDiffusionPAGImg2ImgPipeline",
469
            "StableDiffusionPAGInpaintPipeline",
470
            "StableDiffusionPAGPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
471
472
473
474
475
476
477
478
479
480
            "StableDiffusionPanoramaPipeline",
            "StableDiffusionParadigmsPipeline",
            "StableDiffusionPipeline",
            "StableDiffusionPipelineSafe",
            "StableDiffusionPix2PixZeroPipeline",
            "StableDiffusionSAGPipeline",
            "StableDiffusionUpscalePipeline",
            "StableDiffusionXLAdapterPipeline",
            "StableDiffusionXLControlNetImg2ImgPipeline",
            "StableDiffusionXLControlNetInpaintPipeline",
481
            "StableDiffusionXLControlNetPAGImg2ImgPipeline",
YiYi Xu's avatar
YiYi Xu committed
482
            "StableDiffusionXLControlNetPAGPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
483
            "StableDiffusionXLControlNetPipeline",
hlky's avatar
hlky committed
484
485
486
            "StableDiffusionXLControlNetUnionImg2ImgPipeline",
            "StableDiffusionXLControlNetUnionInpaintPipeline",
            "StableDiffusionXLControlNetUnionPipeline",
487
            "StableDiffusionXLControlNetXSPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
488
489
490
            "StableDiffusionXLImg2ImgPipeline",
            "StableDiffusionXLInpaintPipeline",
            "StableDiffusionXLInstructPix2PixPipeline",
YiYi Xu's avatar
YiYi Xu committed
491
492
493
            "StableDiffusionXLPAGImg2ImgPipeline",
            "StableDiffusionXLPAGInpaintPipeline",
            "StableDiffusionXLPAGPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
494
495
496
            "StableDiffusionXLPipeline",
            "StableUnCLIPImg2ImgPipeline",
            "StableUnCLIPPipeline",
Suraj Patil's avatar
Suraj Patil committed
497
            "StableVideoDiffusionPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
498
499
            "TextToVideoSDPipeline",
            "TextToVideoZeroPipeline",
500
            "TextToVideoZeroSDXLPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
501
502
503
504
505
506
507
508
509
510
511
            "UnCLIPImageVariationPipeline",
            "UnCLIPPipeline",
            "UniDiffuserModel",
            "UniDiffuserPipeline",
            "UniDiffuserTextDecoder",
            "VersatileDiffusionDualGuidedPipeline",
            "VersatileDiffusionImageVariationPipeline",
            "VersatileDiffusionPipeline",
            "VersatileDiffusionTextToImagePipeline",
            "VideoToVideoSDPipeline",
            "VQDiffusionPipeline",
512
513
            "WanImageToVideoPipeline",
            "WanPipeline",
514
            "WanVideoToVideoPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
515
516
517
518
            "WuerstchenCombinedPipeline",
            "WuerstchenDecoderPipeline",
            "WuerstchenPriorPipeline",
        ]
519
    )
520

Sayak Paul's avatar
Sayak Paul committed
521
522
523
524
525
526
527
528
529
530
531
532
533
try:
    if not (is_torch_available() and is_transformers_available() and is_opencv_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    from .utils import dummy_torch_and_transformers_and_opencv_objects  # noqa F403

    _import_structure["utils.dummy_torch_and_transformers_and_opencv_objects"] = [
        name for name in dir(dummy_torch_and_transformers_and_opencv_objects) if not name.startswith("_")
    ]

else:
    _import_structure["pipelines"].extend(["ConsisIDPipeline"])

534
535
536
537
try:
    if not (is_torch_available() and is_transformers_available() and is_k_diffusion_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
538
539
540
541
542
543
    from .utils import dummy_torch_and_transformers_and_k_diffusion_objects  # noqa F403

    _import_structure["utils.dummy_torch_and_transformers_and_k_diffusion_objects"] = [
        name for name in dir(dummy_torch_and_transformers_and_k_diffusion_objects) if not name.startswith("_")
    ]

544
else:
545
    _import_structure["pipelines"].extend(["StableDiffusionKDiffusionPipeline", "StableDiffusionXLKDiffusionPipeline"])
546

547
548
549
550
551
552
553
554
555
556
557
try:
    if not (is_torch_available() and is_transformers_available() and is_sentencepiece_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    from .utils import dummy_torch_and_transformers_and_sentencepiece_objects  # noqa F403

    _import_structure["utils.dummy_torch_and_transformers_and_sentencepiece_objects"] = [
        name for name in dir(dummy_torch_and_transformers_and_sentencepiece_objects) if not name.startswith("_")
    ]

else:
Álvaro Somoza's avatar
Álvaro Somoza committed
558
    _import_structure["pipelines"].extend(["KolorsImg2ImgPipeline", "KolorsPAGPipeline", "KolorsPipeline"])
559

560
561
562
563
try:
    if not (is_torch_available() and is_transformers_available() and is_onnx_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
564
565
566
567
568
569
    from .utils import dummy_torch_and_transformers_and_onnx_objects  # noqa F403

    _import_structure["utils.dummy_torch_and_transformers_and_onnx_objects"] = [
        name for name in dir(dummy_torch_and_transformers_and_onnx_objects) if not name.startswith("_")
    ]

570
else:
Dhruv Nair's avatar
Dhruv Nair committed
571
572
573
574
575
576
577
578
579
    _import_structure["pipelines"].extend(
        [
            "OnnxStableDiffusionImg2ImgPipeline",
            "OnnxStableDiffusionInpaintPipeline",
            "OnnxStableDiffusionInpaintPipelineLegacy",
            "OnnxStableDiffusionPipeline",
            "OnnxStableDiffusionUpscalePipeline",
            "StableDiffusionOnnxPipeline",
        ]
580
    )
581
582
583
584
585

try:
    if not (is_torch_available() and is_librosa_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
586
587
588
589
590
591
    from .utils import dummy_torch_and_librosa_objects  # noqa F403

    _import_structure["utils.dummy_torch_and_librosa_objects"] = [
        name for name in dir(dummy_torch_and_librosa_objects) if not name.startswith("_")
    ]

592
else:
Dhruv Nair's avatar
Dhruv Nair committed
593
    _import_structure["pipelines"].extend(["AudioDiffusionPipeline", "Mel"])
594

595
try:
596
    if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
597
598
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
599
600
601
602
603
604
605
    from .utils import dummy_transformers_and_torch_and_note_seq_objects  # noqa F403

    _import_structure["utils.dummy_transformers_and_torch_and_note_seq_objects"] = [
        name for name in dir(dummy_transformers_and_torch_and_note_seq_objects) if not name.startswith("_")
    ]


606
else:
Dhruv Nair's avatar
Dhruv Nair committed
607
    _import_structure["pipelines"].extend(["SpectrogramDiffusionPipeline"])
608

609
610
611
612
try:
    if not is_flax_available():
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
613
614
615
616
617
618
619
    from .utils import dummy_flax_objects  # noqa F403

    _import_structure["utils.dummy_flax_objects"] = [
        name for name in dir(dummy_flax_objects) if not name.startswith("_")
    ]


620
else:
621
    _import_structure["models.controlnets.controlnet_flax"] = ["FlaxControlNetModel"]
Dhruv Nair's avatar
Dhruv Nair committed
622
    _import_structure["models.modeling_flax_utils"] = ["FlaxModelMixin"]
623
    _import_structure["models.unets.unet_2d_condition_flax"] = ["FlaxUNet2DConditionModel"]
Dhruv Nair's avatar
Dhruv Nair committed
624
625
626
627
628
629
630
    _import_structure["models.vae_flax"] = ["FlaxAutoencoderKL"]
    _import_structure["pipelines"].extend(["FlaxDiffusionPipeline"])
    _import_structure["schedulers"].extend(
        [
            "FlaxDDIMScheduler",
            "FlaxDDPMScheduler",
            "FlaxDPMSolverMultistepScheduler",
Pedro Cuenca's avatar
Pedro Cuenca committed
631
            "FlaxEulerDiscreteScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
632
633
634
635
636
637
            "FlaxKarrasVeScheduler",
            "FlaxLMSDiscreteScheduler",
            "FlaxPNDMScheduler",
            "FlaxSchedulerMixin",
            "FlaxScoreSdeVeScheduler",
        ]
638
639
    )

640

641
642
643
644
try:
    if not (is_flax_available() and is_transformers_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
645
646
647
648
649
650
651
    from .utils import dummy_flax_and_transformers_objects  # noqa F403

    _import_structure["utils.dummy_flax_and_transformers_objects"] = [
        name for name in dir(dummy_flax_and_transformers_objects) if not name.startswith("_")
    ]


652
else:
Dhruv Nair's avatar
Dhruv Nair committed
653
654
655
656
657
658
    _import_structure["pipelines"].extend(
        [
            "FlaxStableDiffusionControlNetPipeline",
            "FlaxStableDiffusionImg2ImgPipeline",
            "FlaxStableDiffusionInpaintPipeline",
            "FlaxStableDiffusionPipeline",
Pedro Cuenca's avatar
Pedro Cuenca committed
659
            "FlaxStableDiffusionXLPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
660
        ]
661
    )
662
663
664
665
666

try:
    if not (is_note_seq_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
667
668
669
670
671
672
673
    from .utils import dummy_note_seq_objects  # noqa F403

    _import_structure["utils.dummy_note_seq_objects"] = [
        name for name in dir(dummy_note_seq_objects) if not name.startswith("_")
    ]


674
else:
Dhruv Nair's avatar
Dhruv Nair committed
675
676
    _import_structure["pipelines"].extend(["MidiProcessor"])

677
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
Dhruv Nair's avatar
Dhruv Nair committed
678
    from .configuration_utils import ConfigMixin
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710

    try:
        if not is_bitsandbytes_available():
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_bitsandbytes_objects import *
    else:
        from .quantizers.quantization_config import BitsAndBytesConfig

    try:
        if not is_gguf_available():
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_gguf_objects import *
    else:
        from .quantizers.quantization_config import GGUFQuantizationConfig

    try:
        if not is_torchao_available():
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_torchao_objects import *
    else:
        from .quantizers.quantization_config import TorchAoConfig

    try:
        if not is_optimum_quanto_available():
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_optimum_quanto_objects import *
    else:
        from .quantizers.quantization_config import QuantoConfig
Dhruv Nair's avatar
Dhruv Nair committed
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725

    try:
        if not is_onnx_available():
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_onnx_objects import *  # noqa F403
    else:
        from .pipelines import OnnxRuntimeModel

    try:
        if not is_torch_available():
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_pt_objects import *  # noqa F403
    else:
Aryan's avatar
Aryan committed
726
727
728
729
730
731
732
        from .hooks import (
            FasterCacheConfig,
            HookRegistry,
            PyramidAttentionBroadcastConfig,
            apply_faster_cache,
            apply_pyramid_attention_broadcast,
        )
Dhruv Nair's avatar
Dhruv Nair committed
733
        from .models import (
Aryan's avatar
Aryan committed
734
            AllegroTransformer3DModel,
Dhruv Nair's avatar
Dhruv Nair committed
735
            AsymmetricAutoencoderKL,
Sayak Paul's avatar
Sayak Paul committed
736
            AuraFlowTransformer2DModel,
737
            AutoencoderDC,
Dhruv Nair's avatar
Dhruv Nair committed
738
            AutoencoderKL,
Aryan's avatar
Aryan committed
739
            AutoencoderKLAllegro,
740
            AutoencoderKLCogVideoX,
Aryan's avatar
Aryan committed
741
            AutoencoderKLHunyuanVideo,
Aryan's avatar
Aryan committed
742
            AutoencoderKLLTXVideo,
743
            AutoencoderKLMagvit,
Aryan's avatar
Aryan committed
744
            AutoencoderKLMochi,
Suraj Patil's avatar
Suraj Patil committed
745
            AutoencoderKLTemporalDecoder,
746
            AutoencoderKLWan,
747
            AutoencoderOobleck,
Dhruv Nair's avatar
Dhruv Nair committed
748
            AutoencoderTiny,
hlky's avatar
hlky committed
749
            AutoModel,
750
            CacheMixin,
751
            CogVideoXTransformer3DModel,
Yuxuan.Zhang's avatar
Yuxuan.Zhang committed
752
            CogView3PlusTransformer2DModel,
753
            CogView4Transformer2DModel,
Shenghai Yuan's avatar
Shenghai Yuan committed
754
            ConsisIDTransformer3DModel,
Will Berman's avatar
Will Berman committed
755
            ConsistencyDecoderVAE,
Dhruv Nair's avatar
Dhruv Nair committed
756
            ControlNetModel,
hlky's avatar
hlky committed
757
            ControlNetUnionModel,
758
            ControlNetXSAdapter,
759
            DiTTransformer2DModel,
760
            EasyAnimateTransformer3DModel,
王奇勋's avatar
王奇勋 committed
761
            FluxControlNetModel,
762
            FluxMultiControlNetModel,
Sayak Paul's avatar
Sayak Paul committed
763
            FluxTransformer2DModel,
764
            HunyuanDiT2DControlNetModel,
765
            HunyuanDiT2DModel,
766
            HunyuanDiT2DMultiControlNetModel,
Aryan's avatar
Aryan committed
767
            HunyuanVideoTransformer3DModel,
768
            I2VGenXLUNet,
769
            Kandinsky3UNet,
770
            LatteTransformer3DModel,
Aryan's avatar
Aryan committed
771
            LTXVideoTransformer3DModel,
Le Zhuo's avatar
Le Zhuo committed
772
            Lumina2Transformer2DModel,
773
            LuminaNextDiT2DModel,
Aryan's avatar
Aryan committed
774
            MochiTransformer3DModel,
Dhruv Nair's avatar
Dhruv Nair committed
775
            ModelMixin,
Dhruv Nair's avatar
Dhruv Nair committed
776
            MotionAdapter,
Dhruv Nair's avatar
Dhruv Nair committed
777
            MultiAdapter,
778
            MultiControlNetModel,
Shitao Xiao's avatar
Shitao Xiao committed
779
            OmniGenTransformer2DModel,
780
            PixArtTransformer2DModel,
Dhruv Nair's avatar
Dhruv Nair committed
781
            PriorTransformer,
782
            SanaTransformer2DModel,
783
784
            SD3ControlNetModel,
            SD3MultiControlNetModel,
Dhruv Nair's avatar
Dhruv Nair committed
785
            SD3Transformer2DModel,
Aryan's avatar
Aryan committed
786
            SparseControlNetModel,
787
            StableAudioDiTModel,
Dhruv Nair's avatar
Dhruv Nair committed
788
789
790
            T2IAdapter,
            T5FilmDecoder,
            Transformer2DModel,
hlky's avatar
hlky committed
791
            TransformerTemporalModel,
Dhruv Nair's avatar
Dhruv Nair committed
792
793
794
795
            UNet1DModel,
            UNet2DConditionModel,
            UNet2DModel,
            UNet3DConditionModel,
796
            UNetControlNetXSModel,
Dhruv Nair's avatar
Dhruv Nair committed
797
            UNetMotionModel,
Suraj Patil's avatar
Suraj Patil committed
798
            UNetSpatioTemporalConditionModel,
Will Berman's avatar
Will Berman committed
799
            UVit2DModel,
Dhruv Nair's avatar
Dhruv Nair committed
800
            VQModel,
801
            WanTransformer3DModel,
Dhruv Nair's avatar
Dhruv Nair committed
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
        )
        from .optimization import (
            get_constant_schedule,
            get_constant_schedule_with_warmup,
            get_cosine_schedule_with_warmup,
            get_cosine_with_hard_restarts_schedule_with_warmup,
            get_linear_schedule_with_warmup,
            get_polynomial_decay_schedule_with_warmup,
            get_scheduler,
        )
        from .pipelines import (
            AudioPipelineOutput,
            AutoPipelineForImage2Image,
            AutoPipelineForInpainting,
            AutoPipelineForText2Image,
Ayush Mangal's avatar
Ayush Mangal committed
817
818
            BlipDiffusionControlNetPipeline,
            BlipDiffusionPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
819
820
821
822
823
824
825
826
827
828
829
830
831
832
            CLIPImageProjection,
            ConsistencyModelPipeline,
            DanceDiffusionPipeline,
            DDIMPipeline,
            DDPMPipeline,
            DiffusionPipeline,
            DiTPipeline,
            ImagePipelineOutput,
            KarrasVePipeline,
            LDMPipeline,
            LDMSuperResolutionPipeline,
            PNDMPipeline,
            RePaintPipeline,
            ScoreSdeVePipeline,
833
            StableDiffusionMixin,
Dhruv Nair's avatar
Dhruv Nair committed
834
        )
835
        from .quantizers import DiffusersQuantizer
Dhruv Nair's avatar
Dhruv Nair committed
836
        from .schedulers import (
Will Berman's avatar
Will Berman committed
837
            AmusedScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
838
            CMStochasticIterativeScheduler,
839
840
            CogVideoXDDIMScheduler,
            CogVideoXDPMScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
841
842
843
844
845
846
847
848
849
850
            DDIMInverseScheduler,
            DDIMParallelScheduler,
            DDIMScheduler,
            DDPMParallelScheduler,
            DDPMScheduler,
            DDPMWuerstchenScheduler,
            DEISMultistepScheduler,
            DPMSolverMultistepInverseScheduler,
            DPMSolverMultistepScheduler,
            DPMSolverSinglestepScheduler,
851
            EDMDPMSolverMultistepScheduler,
Suraj Patil's avatar
Suraj Patil committed
852
            EDMEulerScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
853
854
            EulerAncestralDiscreteScheduler,
            EulerDiscreteScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
855
            FlowMatchEulerDiscreteScheduler,
856
            FlowMatchHeunDiscreteScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
857
858
859
860
861
            HeunDiscreteScheduler,
            IPNDMScheduler,
            KarrasVeScheduler,
            KDPM2AncestralDiscreteScheduler,
            KDPM2DiscreteScheduler,
862
            LCMScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
863
864
            PNDMScheduler,
            RePaintScheduler,
Junsong Chen's avatar
Junsong Chen committed
865
            SASolverScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
866
            SchedulerMixin,
YiYi Xu's avatar
YiYi Xu committed
867
            SCMScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
868
            ScoreSdeVeScheduler,
Michael's avatar
Michael committed
869
            TCDScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
            UnCLIPScheduler,
            UniPCMultistepScheduler,
            VQDiffusionScheduler,
        )
        from .training_utils import EMAModel

    try:
        if not (is_torch_available() and is_scipy_available()):
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_torch_and_scipy_objects import *  # noqa F403
    else:
        from .schedulers import LMSDiscreteScheduler

    try:
        if not (is_torch_available() and is_torchsde_available()):
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_torch_and_torchsde_objects import *  # noqa F403
    else:
890
        from .schedulers import CosineDPMSolverMultistepScheduler, DPMSolverSDEScheduler
Dhruv Nair's avatar
Dhruv Nair committed
891
892
893
894
895
896
897
898

    try:
        if not (is_torch_available() and is_transformers_available()):
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_torch_and_transformers_objects import *  # noqa F403
    else:
        from .pipelines import (
Aryan's avatar
Aryan committed
899
            AllegroPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
900
901
            AltDiffusionImg2ImgPipeline,
            AltDiffusionPipeline,
Will Berman's avatar
Will Berman committed
902
903
904
            AmusedImg2ImgPipeline,
            AmusedInpaintPipeline,
            AmusedPipeline,
905
            AnimateDiffControlNetPipeline,
Aryan's avatar
Aryan committed
906
            AnimateDiffPAGPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
907
            AnimateDiffPipeline,
Aryan's avatar
Aryan committed
908
            AnimateDiffSDXLPipeline,
Aryan's avatar
Aryan committed
909
            AnimateDiffSparseControlNetPipeline,
910
            AnimateDiffVideoToVideoControlNetPipeline,
Aryan V S's avatar
Aryan V S committed
911
            AnimateDiffVideoToVideoPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
912
913
914
915
            AudioLDM2Pipeline,
            AudioLDM2ProjectionModel,
            AudioLDM2UNet2DConditionModel,
            AudioLDMPipeline,
Sayak Paul's avatar
Sayak Paul committed
916
            AuraFlowPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
917
            CLIPImageProjection,
918
            CogVideoXFunControlPipeline,
919
            CogVideoXImageToVideoPipeline,
920
            CogVideoXPipeline,
921
            CogVideoXVideoToVideoPipeline,
Yuxuan.Zhang's avatar
Yuxuan.Zhang committed
922
            CogView3PlusPipeline,
Yuxuan Zhang's avatar
Yuxuan Zhang committed
923
            CogView4ControlPipeline,
924
            CogView4Pipeline,
Dhruv Nair's avatar
Dhruv Nair committed
925
            CycleDiffusionPipeline,
926
927
928
            EasyAnimateControlPipeline,
            EasyAnimateInpaintPipeline,
            EasyAnimatePipeline,
Aryan's avatar
Aryan committed
929
            FluxControlImg2ImgPipeline,
930
            FluxControlInpaintPipeline,
931
932
            FluxControlNetImg2ImgPipeline,
            FluxControlNetInpaintPipeline,
王奇勋's avatar
王奇勋 committed
933
            FluxControlNetPipeline,
Aryan's avatar
Aryan committed
934
935
            FluxControlPipeline,
            FluxFillPipeline,
936
937
            FluxImg2ImgPipeline,
            FluxInpaintPipeline,
Sayak Paul's avatar
Sayak Paul committed
938
            FluxPipeline,
Aryan's avatar
Aryan committed
939
            FluxPriorReduxPipeline,
940
            HunyuanDiTControlNetPipeline,
941
            HunyuanDiTPAGPipeline,
942
            HunyuanDiTPipeline,
Aryan's avatar
Aryan committed
943
            HunyuanSkyreelsImageToVideoPipeline,
Aryan's avatar
Aryan committed
944
            HunyuanVideoImageToVideoPipeline,
Aryan's avatar
Aryan committed
945
            HunyuanVideoPipeline,
946
            I2VGenXLPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
947
948
949
950
951
952
953
            IFImg2ImgPipeline,
            IFImg2ImgSuperResolutionPipeline,
            IFInpaintingPipeline,
            IFInpaintingSuperResolutionPipeline,
            IFPipeline,
            IFSuperResolutionPipeline,
            ImageTextPipelineOutput,
954
955
            Kandinsky3Img2ImgPipeline,
            Kandinsky3Pipeline,
Dhruv Nair's avatar
Dhruv Nair committed
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
            KandinskyCombinedPipeline,
            KandinskyImg2ImgCombinedPipeline,
            KandinskyImg2ImgPipeline,
            KandinskyInpaintCombinedPipeline,
            KandinskyInpaintPipeline,
            KandinskyPipeline,
            KandinskyPriorPipeline,
            KandinskyV22CombinedPipeline,
            KandinskyV22ControlnetImg2ImgPipeline,
            KandinskyV22ControlnetPipeline,
            KandinskyV22Img2ImgCombinedPipeline,
            KandinskyV22Img2ImgPipeline,
            KandinskyV22InpaintCombinedPipeline,
            KandinskyV22InpaintPipeline,
            KandinskyV22Pipeline,
            KandinskyV22PriorEmb2EmbPipeline,
            KandinskyV22PriorPipeline,
973
            LatentConsistencyModelImg2ImgPipeline,
974
            LatentConsistencyModelPipeline,
975
            LattePipeline,
Dhruv Nair's avatar
Dhruv Nair committed
976
            LDMTextToImagePipeline,
977
978
            LEditsPPPipelineStableDiffusion,
            LEditsPPPipelineStableDiffusionXL,
Aryan's avatar
Aryan committed
979
            LTXConditionPipeline,
Aryan's avatar
Aryan committed
980
981
            LTXImageToVideoPipeline,
            LTXPipeline,
982
            Lumina2Pipeline,
Le Zhuo's avatar
Le Zhuo committed
983
            Lumina2Text2ImgPipeline,
984
            LuminaPipeline,
985
            LuminaText2ImgPipeline,
986
            MarigoldDepthPipeline,
987
            MarigoldIntrinsicsPipeline,
988
            MarigoldNormalsPipeline,
Aryan's avatar
Aryan committed
989
            MochiPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
990
            MusicLDMPipeline,
Shitao Xiao's avatar
Shitao Xiao committed
991
            OmniGenPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
992
            PaintByExamplePipeline,
Dhruv Nair's avatar
Dhruv Nair committed
993
            PIAPipeline,
Sayak Paul's avatar
Sayak Paul committed
994
            PixArtAlphaPipeline,
995
            PixArtSigmaPAGPipeline,
996
            PixArtSigmaPipeline,
Aryan's avatar
Aryan committed
997
            ReduxImageEncoder,
998
999
            SanaPAGPipeline,
            SanaPipeline,
YiYi Xu's avatar
YiYi Xu committed
1000
            SanaSprintPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1001
1002
1003
            SemanticStableDiffusionPipeline,
            ShapEImg2ImgPipeline,
            ShapEPipeline,
1004
1005
            StableAudioPipeline,
            StableAudioProjectionModel,
1006
1007
1008
            StableCascadeCombinedPipeline,
            StableCascadeDecoderPipeline,
            StableCascadePriorPipeline,
1009
            StableDiffusion3ControlNetInpaintingPipeline,
1010
            StableDiffusion3ControlNetPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1011
            StableDiffusion3Img2ImgPipeline,
1012
            StableDiffusion3InpaintPipeline,
1013
            StableDiffusion3PAGImg2ImgPipeline,
1014
            StableDiffusion3PAGPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1015
            StableDiffusion3Pipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1016
1017
1018
1019
            StableDiffusionAdapterPipeline,
            StableDiffusionAttendAndExcitePipeline,
            StableDiffusionControlNetImg2ImgPipeline,
            StableDiffusionControlNetInpaintPipeline,
1020
            StableDiffusionControlNetPAGInpaintPipeline,
1021
            StableDiffusionControlNetPAGPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1022
            StableDiffusionControlNetPipeline,
1023
            StableDiffusionControlNetXSPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1024
1025
1026
1027
1028
1029
1030
1031
1032
1033
1034
1035
            StableDiffusionDepth2ImgPipeline,
            StableDiffusionDiffEditPipeline,
            StableDiffusionGLIGENPipeline,
            StableDiffusionGLIGENTextImagePipeline,
            StableDiffusionImageVariationPipeline,
            StableDiffusionImg2ImgPipeline,
            StableDiffusionInpaintPipeline,
            StableDiffusionInpaintPipelineLegacy,
            StableDiffusionInstructPix2PixPipeline,
            StableDiffusionLatentUpscalePipeline,
            StableDiffusionLDM3DPipeline,
            StableDiffusionModelEditingPipeline,
1036
            StableDiffusionPAGImg2ImgPipeline,
1037
            StableDiffusionPAGInpaintPipeline,
1038
            StableDiffusionPAGPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1039
1040
1041
1042
1043
1044
1045
1046
1047
1048
            StableDiffusionPanoramaPipeline,
            StableDiffusionParadigmsPipeline,
            StableDiffusionPipeline,
            StableDiffusionPipelineSafe,
            StableDiffusionPix2PixZeroPipeline,
            StableDiffusionSAGPipeline,
            StableDiffusionUpscalePipeline,
            StableDiffusionXLAdapterPipeline,
            StableDiffusionXLControlNetImg2ImgPipeline,
            StableDiffusionXLControlNetInpaintPipeline,
1049
            StableDiffusionXLControlNetPAGImg2ImgPipeline,
YiYi Xu's avatar
YiYi Xu committed
1050
            StableDiffusionXLControlNetPAGPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1051
            StableDiffusionXLControlNetPipeline,
hlky's avatar
hlky committed
1052
1053
1054
            StableDiffusionXLControlNetUnionImg2ImgPipeline,
            StableDiffusionXLControlNetUnionInpaintPipeline,
            StableDiffusionXLControlNetUnionPipeline,
1055
            StableDiffusionXLControlNetXSPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1056
1057
1058
            StableDiffusionXLImg2ImgPipeline,
            StableDiffusionXLInpaintPipeline,
            StableDiffusionXLInstructPix2PixPipeline,
YiYi Xu's avatar
YiYi Xu committed
1059
1060
1061
            StableDiffusionXLPAGImg2ImgPipeline,
            StableDiffusionXLPAGInpaintPipeline,
            StableDiffusionXLPAGPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1062
1063
1064
            StableDiffusionXLPipeline,
            StableUnCLIPImg2ImgPipeline,
            StableUnCLIPPipeline,
Suraj Patil's avatar
Suraj Patil committed
1065
            StableVideoDiffusionPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1066
1067
            TextToVideoSDPipeline,
            TextToVideoZeroPipeline,
1068
            TextToVideoZeroSDXLPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
            UnCLIPImageVariationPipeline,
            UnCLIPPipeline,
            UniDiffuserModel,
            UniDiffuserPipeline,
            UniDiffuserTextDecoder,
            VersatileDiffusionDualGuidedPipeline,
            VersatileDiffusionImageVariationPipeline,
            VersatileDiffusionPipeline,
            VersatileDiffusionTextToImagePipeline,
            VideoToVideoSDPipeline,
            VQDiffusionPipeline,
1080
1081
            WanImageToVideoPipeline,
            WanPipeline,
1082
            WanVideoToVideoPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1083
1084
1085
1086
1087
1088
1089
1090
1091
1092
1093
            WuerstchenCombinedPipeline,
            WuerstchenDecoderPipeline,
            WuerstchenPriorPipeline,
        )

    try:
        if not (is_torch_available() and is_transformers_available() and is_k_diffusion_available()):
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_torch_and_transformers_and_k_diffusion_objects import *  # noqa F403
    else:
1094
        from .pipelines import StableDiffusionKDiffusionPipeline, StableDiffusionXLKDiffusionPipeline
Dhruv Nair's avatar
Dhruv Nair committed
1095

1096
1097
1098
1099
1100
1101
    try:
        if not (is_torch_available() and is_transformers_available() and is_sentencepiece_available()):
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_torch_and_transformers_and_sentencepiece_objects import *  # noqa F403
    else:
Álvaro Somoza's avatar
Álvaro Somoza committed
1102
        from .pipelines import KolorsImg2ImgPipeline, KolorsPAGPipeline, KolorsPipeline
Sayak Paul's avatar
Sayak Paul committed
1103
1104
1105
1106
1107
1108
1109
1110
1111

    try:
        if not (is_torch_available() and is_transformers_available() and is_opencv_available()):
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_torch_and_transformers_and_opencv_objects import *  # noqa F403
    else:
        from .pipelines import ConsisIDPipeline

Dhruv Nair's avatar
Dhruv Nair committed
1112
1113
1114
1115
1116
1117
1118
1119
1120
1121
1122
1123
1124
1125
1126
1127
1128
1129
1130
1131
1132
1133
1134
1135
1136
1137
1138
1139
1140
1141
1142
1143
1144
1145
1146
1147
1148
    try:
        if not (is_torch_available() and is_transformers_available() and is_onnx_available()):
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_torch_and_transformers_and_onnx_objects import *  # noqa F403
    else:
        from .pipelines import (
            OnnxStableDiffusionImg2ImgPipeline,
            OnnxStableDiffusionInpaintPipeline,
            OnnxStableDiffusionInpaintPipelineLegacy,
            OnnxStableDiffusionPipeline,
            OnnxStableDiffusionUpscalePipeline,
            StableDiffusionOnnxPipeline,
        )

    try:
        if not (is_torch_available() and is_librosa_available()):
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_torch_and_librosa_objects import *  # noqa F403
    else:
        from .pipelines import AudioDiffusionPipeline, Mel

    try:
        if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_transformers_and_torch_and_note_seq_objects import *  # noqa F403
    else:
        from .pipelines import SpectrogramDiffusionPipeline

    try:
        if not is_flax_available():
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_flax_objects import *  # noqa F403
    else:
1149
        from .models.controlnets.controlnet_flax import FlaxControlNetModel
Dhruv Nair's avatar
Dhruv Nair committed
1150
        from .models.modeling_flax_utils import FlaxModelMixin
1151
        from .models.unets.unet_2d_condition_flax import FlaxUNet2DConditionModel
Dhruv Nair's avatar
Dhruv Nair committed
1152
1153
1154
1155
1156
1157
        from .models.vae_flax import FlaxAutoencoderKL
        from .pipelines import FlaxDiffusionPipeline
        from .schedulers import (
            FlaxDDIMScheduler,
            FlaxDDPMScheduler,
            FlaxDPMSolverMultistepScheduler,
Pedro Cuenca's avatar
Pedro Cuenca committed
1158
            FlaxEulerDiscreteScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
1159
1160
1161
1162
1163
1164
1165
1166
1167
1168
1169
1170
1171
1172
1173
1174
1175
1176
            FlaxKarrasVeScheduler,
            FlaxLMSDiscreteScheduler,
            FlaxPNDMScheduler,
            FlaxSchedulerMixin,
            FlaxScoreSdeVeScheduler,
        )

    try:
        if not (is_flax_available() and is_transformers_available()):
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_flax_and_transformers_objects import *  # noqa F403
    else:
        from .pipelines import (
            FlaxStableDiffusionControlNetPipeline,
            FlaxStableDiffusionImg2ImgPipeline,
            FlaxStableDiffusionInpaintPipeline,
            FlaxStableDiffusionPipeline,
Pedro Cuenca's avatar
Pedro Cuenca committed
1177
            FlaxStableDiffusionXLPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1178
1179
1180
1181
1182
1183
1184
1185
1186
1187
1188
1189
1190
1191
1192
1193
1194
1195
1196
1197
        )

    try:
        if not (is_note_seq_available()):
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        from .utils.dummy_note_seq_objects import *  # noqa F403
    else:
        from .pipelines import MidiProcessor

else:
    import sys

    sys.modules[__name__] = _LazyModule(
        __name__,
        globals()["__file__"],
        _import_structure,
        module_spec=__spec__,
        extra_objects={"__version__": __version__},
    )