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

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

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

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

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

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

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

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

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

566
else:
567
    _import_structure["pipelines"].extend(["StableDiffusionKDiffusionPipeline", "StableDiffusionXLKDiffusionPipeline"])
568

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

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

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

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

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

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


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

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


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

662

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


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

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


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

699
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
Dhruv Nair's avatar
Dhruv Nair committed
700
    from .configuration_utils import ConfigMixin
701
    from .quantizers import PipelineQuantizationConfig
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
732
733

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

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

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

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

    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
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
1190
1191
1192
1193
    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:
1194
        from .models.controlnets.controlnet_flax import FlaxControlNetModel
Dhruv Nair's avatar
Dhruv Nair committed
1195
        from .models.modeling_flax_utils import FlaxModelMixin
1196
        from .models.unets.unet_2d_condition_flax import FlaxUNet2DConditionModel
Dhruv Nair's avatar
Dhruv Nair committed
1197
1198
1199
1200
1201
1202
        from .models.vae_flax import FlaxAutoencoderKL
        from .pipelines import FlaxDiffusionPipeline
        from .schedulers import (
            FlaxDDIMScheduler,
            FlaxDDPMScheduler,
            FlaxDPMSolverMultistepScheduler,
Pedro Cuenca's avatar
Pedro Cuenca committed
1203
            FlaxEulerDiscreteScheduler,
Dhruv Nair's avatar
Dhruv Nair committed
1204
1205
1206
1207
1208
1209
1210
1211
1212
1213
1214
1215
1216
1217
1218
1219
1220
1221
            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
1222
            FlaxStableDiffusionXLPipeline,
Dhruv Nair's avatar
Dhruv Nair committed
1223
1224
1225
1226
1227
1228
1229
1230
1231
1232
1233
1234
1235
1236
1237
1238
1239
1240
1241
1242
        )

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