__init__.py 45.3 KB
Newer Older
Sayak Paul's avatar
Sayak Paul committed
1
__version__ = "0.34.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
            "AutoencoderKLCosmos",
Aryan's avatar
Aryan committed
152
            "AutoencoderKLHunyuanVideo",
Aryan's avatar
Aryan committed
153
            "AutoencoderKLLTXVideo",
154
            "AutoencoderKLMagvit",
Aryan's avatar
Aryan committed
155
            "AutoencoderKLMochi",
Suraj Patil's avatar
Suraj Patil committed
156
            "AutoencoderKLTemporalDecoder",
157
            "AutoencoderKLWan",
158
            "AutoencoderOobleck",
Dhruv Nair's avatar
Dhruv Nair committed
159
            "AutoencoderTiny",
hlky's avatar
hlky committed
160
            "AutoModel",
161
            "CacheMixin",
162
            "CogVideoXTransformer3DModel",
Yuxuan.Zhang's avatar
Yuxuan.Zhang committed
163
            "CogView3PlusTransformer2DModel",
164
            "CogView4Transformer2DModel",
Shenghai Yuan's avatar
Shenghai Yuan committed
165
            "ConsisIDTransformer3DModel",
Will Berman's avatar
Will Berman committed
166
            "ConsistencyDecoderVAE",
Dhruv Nair's avatar
Dhruv Nair committed
167
            "ControlNetModel",
hlky's avatar
hlky committed
168
            "ControlNetUnionModel",
169
            "ControlNetXSAdapter",
Aryan's avatar
Aryan committed
170
            "CosmosTransformer3DModel",
171
            "DiTTransformer2DModel",
172
            "EasyAnimateTransformer3DModel",
王奇勋's avatar
王奇勋 committed
173
            "FluxControlNetModel",
174
            "FluxMultiControlNetModel",
Sayak Paul's avatar
Sayak Paul committed
175
            "FluxTransformer2DModel",
hlky's avatar
hlky committed
176
            "HiDreamImageTransformer2DModel",
177
            "HunyuanDiT2DControlNetModel",
178
            "HunyuanDiT2DModel",
179
            "HunyuanDiT2DMultiControlNetModel",
Aryan's avatar
Aryan committed
180
            "HunyuanVideoFramepackTransformer3DModel",
Aryan's avatar
Aryan committed
181
            "HunyuanVideoTransformer3DModel",
182
            "I2VGenXLUNet",
183
            "Kandinsky3UNet",
184
            "LatteTransformer3DModel",
Aryan's avatar
Aryan committed
185
            "LTXVideoTransformer3DModel",
Le Zhuo's avatar
Le Zhuo committed
186
            "Lumina2Transformer2DModel",
187
            "LuminaNextDiT2DModel",
Aryan's avatar
Aryan committed
188
            "MochiTransformer3DModel",
Dhruv Nair's avatar
Dhruv Nair committed
189
            "ModelMixin",
Dhruv Nair's avatar
Dhruv Nair committed
190
            "MotionAdapter",
Dhruv Nair's avatar
Dhruv Nair committed
191
            "MultiAdapter",
192
            "MultiControlNetModel",
Shitao Xiao's avatar
Shitao Xiao committed
193
            "OmniGenTransformer2DModel",
194
            "PixArtTransformer2DModel",
Dhruv Nair's avatar
Dhruv Nair committed
195
            "PriorTransformer",
196
            "SanaControlNetModel",
197
            "SanaTransformer2DModel",
198
199
            "SD3ControlNetModel",
            "SD3MultiControlNetModel",
Dhruv Nair's avatar
Dhruv Nair committed
200
            "SD3Transformer2DModel",
Aryan's avatar
Aryan committed
201
            "SparseControlNetModel",
202
            "StableAudioDiTModel",
203
            "StableCascadeUNet",
Dhruv Nair's avatar
Dhruv Nair committed
204
205
206
            "T2IAdapter",
            "T5FilmDecoder",
            "Transformer2DModel",
hlky's avatar
hlky committed
207
            "TransformerTemporalModel",
Dhruv Nair's avatar
Dhruv Nair committed
208
209
210
211
            "UNet1DModel",
            "UNet2DConditionModel",
            "UNet2DModel",
            "UNet3DConditionModel",
212
            "UNetControlNetXSModel",
Dhruv Nair's avatar
Dhruv Nair committed
213
            "UNetMotionModel",
Suraj Patil's avatar
Suraj Patil committed
214
            "UNetSpatioTemporalConditionModel",
Will Berman's avatar
Will Berman committed
215
            "UVit2DModel",
Dhruv Nair's avatar
Dhruv Nair committed
216
            "VQModel",
217
            "WanTransformer3DModel",
Aryan's avatar
Aryan committed
218
            "WanVACETransformer3DModel",
Dhruv Nair's avatar
Dhruv Nair committed
219
        ]
220
    )
Dhruv Nair's avatar
Dhruv Nair committed
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
    _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",
249
            "StableDiffusionMixin",
Dhruv Nair's avatar
Dhruv Nair committed
250
        ]
251
    )
252
    _import_structure["quantizers"] = ["DiffusersQuantizer"]
Dhruv Nair's avatar
Dhruv Nair committed
253
254
    _import_structure["schedulers"].extend(
        [
Will Berman's avatar
Will Berman committed
255
            "AmusedScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
256
            "CMStochasticIterativeScheduler",
257
258
            "CogVideoXDDIMScheduler",
            "CogVideoXDPMScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
259
260
261
262
263
264
265
266
267
268
            "DDIMInverseScheduler",
            "DDIMParallelScheduler",
            "DDIMScheduler",
            "DDPMParallelScheduler",
            "DDPMScheduler",
            "DDPMWuerstchenScheduler",
            "DEISMultistepScheduler",
            "DPMSolverMultistepInverseScheduler",
            "DPMSolverMultistepScheduler",
            "DPMSolverSinglestepScheduler",
269
            "EDMDPMSolverMultistepScheduler",
Suraj Patil's avatar
Suraj Patil committed
270
            "EDMEulerScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
271
272
            "EulerAncestralDiscreteScheduler",
            "EulerDiscreteScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
273
            "FlowMatchEulerDiscreteScheduler",
274
            "FlowMatchHeunDiscreteScheduler",
275
            "FlowMatchLCMScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
276
277
278
279
280
            "HeunDiscreteScheduler",
            "IPNDMScheduler",
            "KarrasVeScheduler",
            "KDPM2AncestralDiscreteScheduler",
            "KDPM2DiscreteScheduler",
281
            "LCMScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
282
283
            "PNDMScheduler",
            "RePaintScheduler",
Junsong Chen's avatar
Junsong Chen committed
284
            "SASolverScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
285
            "SchedulerMixin",
YiYi Xu's avatar
YiYi Xu committed
286
            "SCMScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
287
            "ScoreSdeVeScheduler",
Michael's avatar
Michael committed
288
            "TCDScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
289
290
291
292
            "UnCLIPScheduler",
            "UniPCMultistepScheduler",
            "VQDiffusionScheduler",
        ]
293
    )
Dhruv Nair's avatar
Dhruv Nair committed
294
    _import_structure["training_utils"] = ["EMAModel"]
Patrick von Platen's avatar
Patrick von Platen committed
295

296
297
298
299
try:
    if not (is_torch_available() and is_scipy_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
300
301
302
303
304
305
    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("_")
    ]

306
else:
Dhruv Nair's avatar
Dhruv Nair committed
307
    _import_structure["schedulers"].extend(["LMSDiscreteScheduler"])
Patrick von Platen's avatar
Patrick von Platen committed
308

309
310
311
312
try:
    if not (is_torch_available() and is_torchsde_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
313
314
315
316
317
318
    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("_")
    ]

319
else:
320
    _import_structure["schedulers"].extend(["CosineDPMSolverMultistepScheduler", "DPMSolverSDEScheduler"])
321
322
323
324
325

try:
    if not (is_torch_available() and is_transformers_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
326
327
328
329
330
331
    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("_")
    ]

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

Sayak Paul's avatar
Sayak Paul committed
541
542
543
544
545
546
547
548
549
550
551
552
553
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"])

554
555
556
557
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
558
559
560
561
562
563
    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("_")
    ]

564
else:
565
    _import_structure["pipelines"].extend(["StableDiffusionKDiffusionPipeline", "StableDiffusionXLKDiffusionPipeline"])
566

567
568
569
570
571
572
573
574
575
576
577
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
578
    _import_structure["pipelines"].extend(["KolorsImg2ImgPipeline", "KolorsPAGPipeline", "KolorsPipeline"])
579

580
581
582
583
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
584
585
586
587
588
589
    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("_")
    ]

590
else:
Dhruv Nair's avatar
Dhruv Nair committed
591
592
593
594
595
596
597
598
599
    _import_structure["pipelines"].extend(
        [
            "OnnxStableDiffusionImg2ImgPipeline",
            "OnnxStableDiffusionInpaintPipeline",
            "OnnxStableDiffusionInpaintPipelineLegacy",
            "OnnxStableDiffusionPipeline",
            "OnnxStableDiffusionUpscalePipeline",
            "StableDiffusionOnnxPipeline",
        ]
600
    )
601
602
603
604
605

try:
    if not (is_torch_available() and is_librosa_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
606
607
608
609
610
611
    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("_")
    ]

612
else:
Dhruv Nair's avatar
Dhruv Nair committed
613
    _import_structure["pipelines"].extend(["AudioDiffusionPipeline", "Mel"])
614

615
try:
616
    if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
617
618
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
619
620
621
622
623
624
625
    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("_")
    ]


626
else:
Dhruv Nair's avatar
Dhruv Nair committed
627
    _import_structure["pipelines"].extend(["SpectrogramDiffusionPipeline"])
628

629
630
631
632
try:
    if not is_flax_available():
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
633
634
635
636
637
638
639
    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("_")
    ]


640
else:
641
    _import_structure["models.controlnets.controlnet_flax"] = ["FlaxControlNetModel"]
Dhruv Nair's avatar
Dhruv Nair committed
642
    _import_structure["models.modeling_flax_utils"] = ["FlaxModelMixin"]
643
    _import_structure["models.unets.unet_2d_condition_flax"] = ["FlaxUNet2DConditionModel"]
Dhruv Nair's avatar
Dhruv Nair committed
644
645
646
647
648
649
650
    _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
651
            "FlaxEulerDiscreteScheduler",
Dhruv Nair's avatar
Dhruv Nair committed
652
653
654
655
656
657
            "FlaxKarrasVeScheduler",
            "FlaxLMSDiscreteScheduler",
            "FlaxPNDMScheduler",
            "FlaxSchedulerMixin",
            "FlaxScoreSdeVeScheduler",
        ]
658
659
    )

660

661
662
663
664
try:
    if not (is_flax_available() and is_transformers_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
665
666
667
668
669
670
671
    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("_")
    ]


672
else:
Dhruv Nair's avatar
Dhruv Nair committed
673
674
675
676
677
678
    _import_structure["pipelines"].extend(
        [
            "FlaxStableDiffusionControlNetPipeline",
            "FlaxStableDiffusionImg2ImgPipeline",
            "FlaxStableDiffusionInpaintPipeline",
            "FlaxStableDiffusionPipeline",
Pedro Cuenca's avatar
Pedro Cuenca committed
679
            "FlaxStableDiffusionXLPipeline",
Dhruv Nair's avatar
Dhruv Nair committed
680
        ]
681
    )
682
683
684
685
686

try:
    if not (is_note_seq_available()):
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
Dhruv Nair's avatar
Dhruv Nair committed
687
688
689
690
691
692
693
    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("_")
    ]


694
else:
Dhruv Nair's avatar
Dhruv Nair committed
695
696
    _import_structure["pipelines"].extend(["MidiProcessor"])

697
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
Dhruv Nair's avatar
Dhruv Nair committed
698
    from .configuration_utils import ConfigMixin
699
    from .quantizers import PipelineQuantizationConfig
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731

    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
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746

    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
747
748
749
750
751
752
753
        from .hooks import (
            FasterCacheConfig,
            HookRegistry,
            PyramidAttentionBroadcastConfig,
            apply_faster_cache,
            apply_pyramid_attention_broadcast,
        )
Dhruv Nair's avatar
Dhruv Nair committed
754
        from .models import (
Aryan's avatar
Aryan committed
755
            AllegroTransformer3DModel,
Dhruv Nair's avatar
Dhruv Nair committed
756
            AsymmetricAutoencoderKL,
Sayak Paul's avatar
Sayak Paul committed
757
            AuraFlowTransformer2DModel,
758
            AutoencoderDC,
Dhruv Nair's avatar
Dhruv Nair committed
759
            AutoencoderKL,
Aryan's avatar
Aryan committed
760
            AutoencoderKLAllegro,
761
            AutoencoderKLCogVideoX,
Aryan's avatar
Aryan committed
762
            AutoencoderKLCosmos,
Aryan's avatar
Aryan committed
763
            AutoencoderKLHunyuanVideo,
Aryan's avatar
Aryan committed
764
            AutoencoderKLLTXVideo,
765
            AutoencoderKLMagvit,
Aryan's avatar
Aryan committed
766
            AutoencoderKLMochi,
Suraj Patil's avatar
Suraj Patil committed
767
            AutoencoderKLTemporalDecoder,
768
            AutoencoderKLWan,
769
            AutoencoderOobleck,
Dhruv Nair's avatar
Dhruv Nair committed
770
            AutoencoderTiny,
hlky's avatar
hlky committed
771
            AutoModel,
772
            CacheMixin,
773
            CogVideoXTransformer3DModel,
Yuxuan.Zhang's avatar
Yuxuan.Zhang committed
774
            CogView3PlusTransformer2DModel,
775
            CogView4Transformer2DModel,
Shenghai Yuan's avatar
Shenghai Yuan committed
776
            ConsisIDTransformer3DModel,
Will Berman's avatar
Will Berman committed
777
            ConsistencyDecoderVAE,
Dhruv Nair's avatar
Dhruv Nair committed
778
            ControlNetModel,
hlky's avatar
hlky committed
779
            ControlNetUnionModel,
780
            ControlNetXSAdapter,
Aryan's avatar
Aryan committed
781
            CosmosTransformer3DModel,
782
            DiTTransformer2DModel,
783
            EasyAnimateTransformer3DModel,
王奇勋's avatar
王奇勋 committed
784
            FluxControlNetModel,
785
            FluxMultiControlNetModel,
Sayak Paul's avatar
Sayak Paul committed
786
            FluxTransformer2DModel,
hlky's avatar
hlky committed
787
            HiDreamImageTransformer2DModel,
788
            HunyuanDiT2DControlNetModel,
789
            HunyuanDiT2DModel,
790
            HunyuanDiT2DMultiControlNetModel,
Aryan's avatar
Aryan committed
791
            HunyuanVideoFramepackTransformer3DModel,
Aryan's avatar
Aryan committed
792
            HunyuanVideoTransformer3DModel,
793
            I2VGenXLUNet,
794
            Kandinsky3UNet,
795
            LatteTransformer3DModel,
Aryan's avatar
Aryan committed
796
            LTXVideoTransformer3DModel,
Le Zhuo's avatar
Le Zhuo committed
797
            Lumina2Transformer2DModel,
798
            LuminaNextDiT2DModel,
Aryan's avatar
Aryan committed
799
            MochiTransformer3DModel,
Dhruv Nair's avatar
Dhruv Nair committed
800
            ModelMixin,
Dhruv Nair's avatar
Dhruv Nair committed
801
            MotionAdapter,
Dhruv Nair's avatar
Dhruv Nair committed
802
            MultiAdapter,
803
            MultiControlNetModel,
Shitao Xiao's avatar
Shitao Xiao committed
804
            OmniGenTransformer2DModel,
805
            PixArtTransformer2DModel,
Dhruv Nair's avatar
Dhruv Nair committed
806
            PriorTransformer,
807
            SanaControlNetModel,
808
            SanaTransformer2DModel,
809
810
            SD3ControlNetModel,
            SD3MultiControlNetModel,
Dhruv Nair's avatar
Dhruv Nair committed
811
            SD3Transformer2DModel,
Aryan's avatar
Aryan committed
812
            SparseControlNetModel,
813
            StableAudioDiTModel,
Dhruv Nair's avatar
Dhruv Nair committed
814
815
816
            T2IAdapter,
            T5FilmDecoder,
            Transformer2DModel,
hlky's avatar
hlky committed
817
            TransformerTemporalModel,
Dhruv Nair's avatar
Dhruv Nair committed
818
819
820
821
            UNet1DModel,
            UNet2DConditionModel,
            UNet2DModel,
            UNet3DConditionModel,
822
            UNetControlNetXSModel,
Dhruv Nair's avatar
Dhruv Nair committed
823
            UNetMotionModel,
Suraj Patil's avatar
Suraj Patil committed
824
            UNetSpatioTemporalConditionModel,
Will Berman's avatar
Will Berman committed
825
            UVit2DModel,
Dhruv Nair's avatar
Dhruv Nair committed
826
            VQModel,
827
            WanTransformer3DModel,
Aryan's avatar
Aryan committed
828
            WanVACETransformer3DModel,
Dhruv Nair's avatar
Dhruv Nair committed
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
        )
        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
844
845
            BlipDiffusionControlNetPipeline,
            BlipDiffusionPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
846
847
848
849
850
851
852
853
854
855
856
857
858
859
            CLIPImageProjection,
            ConsistencyModelPipeline,
            DanceDiffusionPipeline,
            DDIMPipeline,
            DDPMPipeline,
            DiffusionPipeline,
            DiTPipeline,
            ImagePipelineOutput,
            KarrasVePipeline,
            LDMPipeline,
            LDMSuperResolutionPipeline,
            PNDMPipeline,
            RePaintPipeline,
            ScoreSdeVePipeline,
860
            StableDiffusionMixin,
Dhruv Nair's avatar
Dhruv Nair committed
861
        )
862
        from .quantizers import DiffusersQuantizer
Dhruv Nair's avatar
Dhruv Nair committed
863
        from .schedulers import (
Will Berman's avatar
Will Berman committed
864
            AmusedScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
865
            CMStochasticIterativeScheduler,
866
867
            CogVideoXDDIMScheduler,
            CogVideoXDPMScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
868
869
870
871
872
873
874
875
876
877
            DDIMInverseScheduler,
            DDIMParallelScheduler,
            DDIMScheduler,
            DDPMParallelScheduler,
            DDPMScheduler,
            DDPMWuerstchenScheduler,
            DEISMultistepScheduler,
            DPMSolverMultistepInverseScheduler,
            DPMSolverMultistepScheduler,
            DPMSolverSinglestepScheduler,
878
            EDMDPMSolverMultistepScheduler,
Suraj Patil's avatar
Suraj Patil committed
879
            EDMEulerScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
880
881
            EulerAncestralDiscreteScheduler,
            EulerDiscreteScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
882
            FlowMatchEulerDiscreteScheduler,
883
            FlowMatchHeunDiscreteScheduler,
884
            FlowMatchLCMScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
885
886
887
888
889
            HeunDiscreteScheduler,
            IPNDMScheduler,
            KarrasVeScheduler,
            KDPM2AncestralDiscreteScheduler,
            KDPM2DiscreteScheduler,
890
            LCMScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
891
892
            PNDMScheduler,
            RePaintScheduler,
Junsong Chen's avatar
Junsong Chen committed
893
            SASolverScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
894
            SchedulerMixin,
YiYi Xu's avatar
YiYi Xu committed
895
            SCMScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
896
            ScoreSdeVeScheduler,
Michael's avatar
Michael committed
897
            TCDScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
            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:
918
        from .schedulers import CosineDPMSolverMultistepScheduler, DPMSolverSDEScheduler
Dhruv Nair's avatar
Dhruv Nair committed
919
920
921
922
923
924
925
926

    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
927
            AllegroPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
928
929
            AltDiffusionImg2ImgPipeline,
            AltDiffusionPipeline,
Will Berman's avatar
Will Berman committed
930
931
932
            AmusedImg2ImgPipeline,
            AmusedInpaintPipeline,
            AmusedPipeline,
933
            AnimateDiffControlNetPipeline,
Aryan's avatar
Aryan committed
934
            AnimateDiffPAGPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
935
            AnimateDiffPipeline,
Aryan's avatar
Aryan committed
936
            AnimateDiffSDXLPipeline,
Aryan's avatar
Aryan committed
937
            AnimateDiffSparseControlNetPipeline,
938
            AnimateDiffVideoToVideoControlNetPipeline,
Aryan V S's avatar
Aryan V S committed
939
            AnimateDiffVideoToVideoPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
940
941
942
943
            AudioLDM2Pipeline,
            AudioLDM2ProjectionModel,
            AudioLDM2UNet2DConditionModel,
            AudioLDMPipeline,
Sayak Paul's avatar
Sayak Paul committed
944
            AuraFlowPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
945
            CLIPImageProjection,
946
            CogVideoXFunControlPipeline,
947
            CogVideoXImageToVideoPipeline,
948
            CogVideoXPipeline,
949
            CogVideoXVideoToVideoPipeline,
Yuxuan.Zhang's avatar
Yuxuan.Zhang committed
950
            CogView3PlusPipeline,
Yuxuan Zhang's avatar
Yuxuan Zhang committed
951
            CogView4ControlPipeline,
952
            CogView4Pipeline,
Aryan's avatar
Aryan committed
953
            ConsisIDPipeline,
Aryan's avatar
Aryan committed
954
955
            Cosmos2TextToImagePipeline,
            Cosmos2VideoToWorldPipeline,
Aryan's avatar
Aryan committed
956
957
            CosmosTextToWorldPipeline,
            CosmosVideoToWorldPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
958
            CycleDiffusionPipeline,
959
960
961
            EasyAnimateControlPipeline,
            EasyAnimateInpaintPipeline,
            EasyAnimatePipeline,
Aryan's avatar
Aryan committed
962
            FluxControlImg2ImgPipeline,
963
            FluxControlInpaintPipeline,
964
965
            FluxControlNetImg2ImgPipeline,
            FluxControlNetInpaintPipeline,
王奇勋's avatar
王奇勋 committed
966
            FluxControlNetPipeline,
Aryan's avatar
Aryan committed
967
968
            FluxControlPipeline,
            FluxFillPipeline,
969
970
            FluxImg2ImgPipeline,
            FluxInpaintPipeline,
Sayak Paul's avatar
Sayak Paul committed
971
            FluxPipeline,
Aryan's avatar
Aryan committed
972
            FluxPriorReduxPipeline,
hlky's avatar
hlky committed
973
            HiDreamImagePipeline,
974
            HunyuanDiTControlNetPipeline,
975
            HunyuanDiTPAGPipeline,
976
            HunyuanDiTPipeline,
Aryan's avatar
Aryan committed
977
            HunyuanSkyreelsImageToVideoPipeline,
Aryan's avatar
Aryan committed
978
            HunyuanVideoFramepackPipeline,
Aryan's avatar
Aryan committed
979
            HunyuanVideoImageToVideoPipeline,
Aryan's avatar
Aryan committed
980
            HunyuanVideoPipeline,
981
            I2VGenXLPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
982
983
984
985
986
987
988
            IFImg2ImgPipeline,
            IFImg2ImgSuperResolutionPipeline,
            IFInpaintingPipeline,
            IFInpaintingSuperResolutionPipeline,
            IFPipeline,
            IFSuperResolutionPipeline,
            ImageTextPipelineOutput,
989
990
            Kandinsky3Img2ImgPipeline,
            Kandinsky3Pipeline,
Dhruv Nair's avatar
Dhruv Nair committed
991
992
993
994
995
996
997
998
999
1000
1001
1002
1003
1004
1005
1006
1007
            KandinskyCombinedPipeline,
            KandinskyImg2ImgCombinedPipeline,
            KandinskyImg2ImgPipeline,
            KandinskyInpaintCombinedPipeline,
            KandinskyInpaintPipeline,
            KandinskyPipeline,
            KandinskyPriorPipeline,
            KandinskyV22CombinedPipeline,
            KandinskyV22ControlnetImg2ImgPipeline,
            KandinskyV22ControlnetPipeline,
            KandinskyV22Img2ImgCombinedPipeline,
            KandinskyV22Img2ImgPipeline,
            KandinskyV22InpaintCombinedPipeline,
            KandinskyV22InpaintPipeline,
            KandinskyV22Pipeline,
            KandinskyV22PriorEmb2EmbPipeline,
            KandinskyV22PriorPipeline,
1008
            LatentConsistencyModelImg2ImgPipeline,
1009
            LatentConsistencyModelPipeline,
1010
            LattePipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1011
            LDMTextToImagePipeline,
1012
1013
            LEditsPPPipelineStableDiffusion,
            LEditsPPPipelineStableDiffusionXL,
Aryan's avatar
Aryan committed
1014
            LTXConditionPipeline,
Aryan's avatar
Aryan committed
1015
            LTXImageToVideoPipeline,
Aryan's avatar
Aryan committed
1016
            LTXLatentUpsamplePipeline,
Aryan's avatar
Aryan committed
1017
            LTXPipeline,
1018
            Lumina2Pipeline,
Le Zhuo's avatar
Le Zhuo committed
1019
            Lumina2Text2ImgPipeline,
1020
            LuminaPipeline,
1021
            LuminaText2ImgPipeline,
1022
            MarigoldDepthPipeline,
1023
            MarigoldIntrinsicsPipeline,
1024
            MarigoldNormalsPipeline,
Aryan's avatar
Aryan committed
1025
            MochiPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1026
            MusicLDMPipeline,
Shitao Xiao's avatar
Shitao Xiao committed
1027
            OmniGenPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1028
            PaintByExamplePipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1029
            PIAPipeline,
Sayak Paul's avatar
Sayak Paul committed
1030
            PixArtAlphaPipeline,
1031
            PixArtSigmaPAGPipeline,
1032
            PixArtSigmaPipeline,
Aryan's avatar
Aryan committed
1033
            ReduxImageEncoder,
1034
            SanaControlNetPipeline,
1035
1036
            SanaPAGPipeline,
            SanaPipeline,
1037
            SanaSprintImg2ImgPipeline,
YiYi Xu's avatar
YiYi Xu committed
1038
            SanaSprintPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1039
1040
1041
            SemanticStableDiffusionPipeline,
            ShapEImg2ImgPipeline,
            ShapEPipeline,
1042
1043
            StableAudioPipeline,
            StableAudioProjectionModel,
1044
1045
1046
            StableCascadeCombinedPipeline,
            StableCascadeDecoderPipeline,
            StableCascadePriorPipeline,
1047
            StableDiffusion3ControlNetInpaintingPipeline,
1048
            StableDiffusion3ControlNetPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1049
            StableDiffusion3Img2ImgPipeline,
1050
            StableDiffusion3InpaintPipeline,
1051
            StableDiffusion3PAGImg2ImgPipeline,
1052
            StableDiffusion3PAGPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1053
            StableDiffusion3Pipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1054
1055
1056
1057
            StableDiffusionAdapterPipeline,
            StableDiffusionAttendAndExcitePipeline,
            StableDiffusionControlNetImg2ImgPipeline,
            StableDiffusionControlNetInpaintPipeline,
1058
            StableDiffusionControlNetPAGInpaintPipeline,
1059
            StableDiffusionControlNetPAGPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1060
            StableDiffusionControlNetPipeline,
1061
            StableDiffusionControlNetXSPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1062
1063
1064
1065
1066
1067
1068
1069
1070
1071
1072
1073
            StableDiffusionDepth2ImgPipeline,
            StableDiffusionDiffEditPipeline,
            StableDiffusionGLIGENPipeline,
            StableDiffusionGLIGENTextImagePipeline,
            StableDiffusionImageVariationPipeline,
            StableDiffusionImg2ImgPipeline,
            StableDiffusionInpaintPipeline,
            StableDiffusionInpaintPipelineLegacy,
            StableDiffusionInstructPix2PixPipeline,
            StableDiffusionLatentUpscalePipeline,
            StableDiffusionLDM3DPipeline,
            StableDiffusionModelEditingPipeline,
1074
            StableDiffusionPAGImg2ImgPipeline,
1075
            StableDiffusionPAGInpaintPipeline,
1076
            StableDiffusionPAGPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1077
1078
1079
1080
1081
1082
1083
1084
1085
1086
            StableDiffusionPanoramaPipeline,
            StableDiffusionParadigmsPipeline,
            StableDiffusionPipeline,
            StableDiffusionPipelineSafe,
            StableDiffusionPix2PixZeroPipeline,
            StableDiffusionSAGPipeline,
            StableDiffusionUpscalePipeline,
            StableDiffusionXLAdapterPipeline,
            StableDiffusionXLControlNetImg2ImgPipeline,
            StableDiffusionXLControlNetInpaintPipeline,
1087
            StableDiffusionXLControlNetPAGImg2ImgPipeline,
YiYi Xu's avatar
YiYi Xu committed
1088
            StableDiffusionXLControlNetPAGPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1089
            StableDiffusionXLControlNetPipeline,
hlky's avatar
hlky committed
1090
1091
1092
            StableDiffusionXLControlNetUnionImg2ImgPipeline,
            StableDiffusionXLControlNetUnionInpaintPipeline,
            StableDiffusionXLControlNetUnionPipeline,
1093
            StableDiffusionXLControlNetXSPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1094
1095
1096
            StableDiffusionXLImg2ImgPipeline,
            StableDiffusionXLInpaintPipeline,
            StableDiffusionXLInstructPix2PixPipeline,
YiYi Xu's avatar
YiYi Xu committed
1097
1098
1099
            StableDiffusionXLPAGImg2ImgPipeline,
            StableDiffusionXLPAGInpaintPipeline,
            StableDiffusionXLPAGPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1100
1101
1102
            StableDiffusionXLPipeline,
            StableUnCLIPImg2ImgPipeline,
            StableUnCLIPPipeline,
Suraj Patil's avatar
Suraj Patil committed
1103
            StableVideoDiffusionPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1104
1105
            TextToVideoSDPipeline,
            TextToVideoZeroPipeline,
1106
            TextToVideoZeroSDXLPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1107
1108
1109
1110
1111
1112
1113
1114
1115
1116
            UnCLIPImageVariationPipeline,
            UnCLIPPipeline,
            UniDiffuserModel,
            UniDiffuserPipeline,
            UniDiffuserTextDecoder,
            VersatileDiffusionDualGuidedPipeline,
            VersatileDiffusionImageVariationPipeline,
            VersatileDiffusionPipeline,
            VersatileDiffusionTextToImagePipeline,
            VideoToVideoSDPipeline,
Zhong-Yu Li's avatar
Zhong-Yu Li committed
1117
1118
            VisualClozeGenerationPipeline,
            VisualClozePipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1119
            VQDiffusionPipeline,
1120
1121
            WanImageToVideoPipeline,
            WanPipeline,
Aryan's avatar
Aryan committed
1122
            WanVACEPipeline,
1123
            WanVideoToVideoPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1124
1125
1126
1127
1128
1129
1130
1131
1132
1133
1134
            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:
1135
        from .pipelines import StableDiffusionKDiffusionPipeline, StableDiffusionXLKDiffusionPipeline
Dhruv Nair's avatar
Dhruv Nair committed
1136

1137
1138
1139
1140
1141
1142
    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
1143
        from .pipelines import KolorsImg2ImgPipeline, KolorsPAGPipeline, KolorsPipeline
Sayak Paul's avatar
Sayak Paul committed
1144
1145
1146
1147
1148
1149
1150
1151
1152

    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
1153
1154
1155
1156
1157
1158
1159
1160
1161
1162
1163
1164
1165
1166
1167
1168
1169
1170
1171
1172
1173
1174
1175
1176
1177
1178
1179
1180
1181
1182
1183
1184
1185
1186
1187
1188
1189
    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:
1190
        from .models.controlnets.controlnet_flax import FlaxControlNetModel
Dhruv Nair's avatar
Dhruv Nair committed
1191
        from .models.modeling_flax_utils import FlaxModelMixin
1192
        from .models.unets.unet_2d_condition_flax import FlaxUNet2DConditionModel
Dhruv Nair's avatar
Dhruv Nair committed
1193
1194
1195
1196
1197
1198
        from .models.vae_flax import FlaxAutoencoderKL
        from .pipelines import FlaxDiffusionPipeline
        from .schedulers import (
            FlaxDDIMScheduler,
            FlaxDDPMScheduler,
            FlaxDPMSolverMultistepScheduler,
Pedro Cuenca's avatar
Pedro Cuenca committed
1199
            FlaxEulerDiscreteScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
1200
1201
1202
1203
1204
1205
1206
1207
1208
1209
1210
1211
1212
1213
1214
1215
1216
1217
            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
1218
            FlaxStableDiffusionXLPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1219
1220
1221
1222
1223
1224
1225
1226
1227
1228
1229
1230
1231
1232
1233
1234
1235
1236
1237
1238
        )

    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__},
    )