diffusers 0.29.2__py3-none-any.whl → 0.30.1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- diffusers/__init__.py +94 -3
- diffusers/commands/env.py +1 -5
- diffusers/configuration_utils.py +4 -9
- diffusers/dependency_versions_table.py +2 -2
- diffusers/image_processor.py +1 -2
- diffusers/loaders/__init__.py +17 -2
- diffusers/loaders/ip_adapter.py +10 -7
- diffusers/loaders/lora_base.py +752 -0
- diffusers/loaders/lora_pipeline.py +2252 -0
- diffusers/loaders/peft.py +213 -5
- diffusers/loaders/single_file.py +3 -14
- diffusers/loaders/single_file_model.py +31 -10
- diffusers/loaders/single_file_utils.py +293 -8
- diffusers/loaders/textual_inversion.py +1 -6
- diffusers/loaders/unet.py +23 -208
- diffusers/models/__init__.py +20 -0
- diffusers/models/activations.py +22 -0
- diffusers/models/attention.py +386 -7
- diffusers/models/attention_processor.py +1937 -629
- diffusers/models/autoencoders/__init__.py +2 -0
- diffusers/models/autoencoders/autoencoder_kl.py +14 -3
- diffusers/models/autoencoders/autoencoder_kl_cogvideox.py +1271 -0
- diffusers/models/autoencoders/autoencoder_kl_temporal_decoder.py +1 -1
- diffusers/models/autoencoders/autoencoder_oobleck.py +464 -0
- diffusers/models/autoencoders/autoencoder_tiny.py +1 -0
- diffusers/models/autoencoders/consistency_decoder_vae.py +1 -1
- diffusers/models/autoencoders/vq_model.py +4 -4
- diffusers/models/controlnet.py +2 -3
- diffusers/models/controlnet_hunyuan.py +401 -0
- diffusers/models/controlnet_sd3.py +11 -11
- diffusers/models/controlnet_sparsectrl.py +789 -0
- diffusers/models/controlnet_xs.py +40 -10
- diffusers/models/downsampling.py +68 -0
- diffusers/models/embeddings.py +403 -36
- diffusers/models/model_loading_utils.py +1 -3
- diffusers/models/modeling_flax_utils.py +1 -6
- diffusers/models/modeling_utils.py +4 -16
- diffusers/models/normalization.py +203 -12
- diffusers/models/transformers/__init__.py +6 -0
- diffusers/models/transformers/auraflow_transformer_2d.py +543 -0
- diffusers/models/transformers/cogvideox_transformer_3d.py +485 -0
- diffusers/models/transformers/hunyuan_transformer_2d.py +19 -15
- diffusers/models/transformers/latte_transformer_3d.py +327 -0
- diffusers/models/transformers/lumina_nextdit2d.py +340 -0
- diffusers/models/transformers/pixart_transformer_2d.py +102 -1
- diffusers/models/transformers/prior_transformer.py +1 -1
- diffusers/models/transformers/stable_audio_transformer.py +458 -0
- diffusers/models/transformers/transformer_flux.py +455 -0
- diffusers/models/transformers/transformer_sd3.py +18 -4
- diffusers/models/unets/unet_1d_blocks.py +1 -1
- diffusers/models/unets/unet_2d_condition.py +8 -1
- diffusers/models/unets/unet_3d_blocks.py +51 -920
- diffusers/models/unets/unet_3d_condition.py +4 -1
- diffusers/models/unets/unet_i2vgen_xl.py +4 -1
- diffusers/models/unets/unet_kandinsky3.py +1 -1
- diffusers/models/unets/unet_motion_model.py +1330 -84
- diffusers/models/unets/unet_spatio_temporal_condition.py +1 -1
- diffusers/models/unets/unet_stable_cascade.py +1 -3
- diffusers/models/unets/uvit_2d.py +1 -1
- diffusers/models/upsampling.py +64 -0
- diffusers/models/vq_model.py +8 -4
- diffusers/optimization.py +1 -1
- diffusers/pipelines/__init__.py +100 -3
- diffusers/pipelines/animatediff/__init__.py +4 -0
- diffusers/pipelines/animatediff/pipeline_animatediff.py +50 -40
- diffusers/pipelines/animatediff/pipeline_animatediff_controlnet.py +1076 -0
- diffusers/pipelines/animatediff/pipeline_animatediff_sdxl.py +17 -27
- diffusers/pipelines/animatediff/pipeline_animatediff_sparsectrl.py +1008 -0
- diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py +51 -38
- diffusers/pipelines/audioldm2/modeling_audioldm2.py +1 -1
- diffusers/pipelines/audioldm2/pipeline_audioldm2.py +1 -0
- diffusers/pipelines/aura_flow/__init__.py +48 -0
- diffusers/pipelines/aura_flow/pipeline_aura_flow.py +591 -0
- diffusers/pipelines/auto_pipeline.py +97 -19
- diffusers/pipelines/cogvideo/__init__.py +48 -0
- diffusers/pipelines/cogvideo/pipeline_cogvideox.py +746 -0
- diffusers/pipelines/consistency_models/pipeline_consistency_models.py +1 -1
- diffusers/pipelines/controlnet/pipeline_controlnet.py +24 -30
- diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py +31 -30
- diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py +24 -153
- diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py +19 -28
- diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py +18 -28
- diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py +29 -32
- diffusers/pipelines/controlnet/pipeline_flax_controlnet.py +2 -2
- diffusers/pipelines/controlnet_hunyuandit/__init__.py +48 -0
- diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py +1042 -0
- diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet.py +35 -0
- diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs.py +10 -6
- diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs_sd_xl.py +0 -4
- diffusers/pipelines/deepfloyd_if/pipeline_if.py +2 -2
- diffusers/pipelines/deepfloyd_if/pipeline_if_img2img.py +2 -2
- diffusers/pipelines/deepfloyd_if/pipeline_if_img2img_superresolution.py +2 -2
- diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting.py +2 -2
- diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting_superresolution.py +2 -2
- diffusers/pipelines/deepfloyd_if/pipeline_if_superresolution.py +2 -2
- diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py +11 -6
- diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py +11 -6
- diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_cycle_diffusion.py +6 -6
- diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_inpaint_legacy.py +6 -6
- diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_model_editing.py +10 -10
- diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_paradigms.py +10 -6
- diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_pix2pix_zero.py +3 -3
- diffusers/pipelines/deprecated/versatile_diffusion/modeling_text_unet.py +1 -1
- diffusers/pipelines/flux/__init__.py +47 -0
- diffusers/pipelines/flux/pipeline_flux.py +749 -0
- diffusers/pipelines/flux/pipeline_output.py +21 -0
- diffusers/pipelines/free_init_utils.py +2 -0
- diffusers/pipelines/free_noise_utils.py +236 -0
- diffusers/pipelines/kandinsky3/pipeline_kandinsky3.py +2 -2
- diffusers/pipelines/kandinsky3/pipeline_kandinsky3_img2img.py +2 -2
- diffusers/pipelines/kolors/__init__.py +54 -0
- diffusers/pipelines/kolors/pipeline_kolors.py +1070 -0
- diffusers/pipelines/kolors/pipeline_kolors_img2img.py +1247 -0
- diffusers/pipelines/kolors/pipeline_output.py +21 -0
- diffusers/pipelines/kolors/text_encoder.py +889 -0
- diffusers/pipelines/kolors/tokenizer.py +334 -0
- diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_img2img.py +30 -29
- diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_text2img.py +23 -29
- diffusers/pipelines/latte/__init__.py +48 -0
- diffusers/pipelines/latte/pipeline_latte.py +881 -0
- diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py +4 -4
- diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py +0 -4
- diffusers/pipelines/lumina/__init__.py +48 -0
- diffusers/pipelines/lumina/pipeline_lumina.py +897 -0
- diffusers/pipelines/pag/__init__.py +67 -0
- diffusers/pipelines/pag/pag_utils.py +237 -0
- diffusers/pipelines/pag/pipeline_pag_controlnet_sd.py +1329 -0
- diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl.py +1612 -0
- diffusers/pipelines/pag/pipeline_pag_hunyuandit.py +953 -0
- diffusers/pipelines/pag/pipeline_pag_kolors.py +1136 -0
- diffusers/pipelines/pag/pipeline_pag_pixart_sigma.py +872 -0
- diffusers/pipelines/pag/pipeline_pag_sd.py +1050 -0
- diffusers/pipelines/pag/pipeline_pag_sd_3.py +985 -0
- diffusers/pipelines/pag/pipeline_pag_sd_animatediff.py +862 -0
- diffusers/pipelines/pag/pipeline_pag_sd_xl.py +1333 -0
- diffusers/pipelines/pag/pipeline_pag_sd_xl_img2img.py +1529 -0
- diffusers/pipelines/pag/pipeline_pag_sd_xl_inpaint.py +1753 -0
- diffusers/pipelines/pia/pipeline_pia.py +30 -37
- diffusers/pipelines/pipeline_flax_utils.py +4 -9
- diffusers/pipelines/pipeline_loading_utils.py +0 -3
- diffusers/pipelines/pipeline_utils.py +2 -14
- diffusers/pipelines/semantic_stable_diffusion/pipeline_semantic_stable_diffusion.py +0 -1
- diffusers/pipelines/stable_audio/__init__.py +50 -0
- diffusers/pipelines/stable_audio/modeling_stable_audio.py +158 -0
- diffusers/pipelines/stable_audio/pipeline_stable_audio.py +745 -0
- diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +2 -0
- diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion.py +1 -1
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py +23 -29
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py +15 -8
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py +30 -29
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py +23 -152
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py +8 -4
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py +11 -11
- diffusers/pipelines/stable_diffusion/pipeline_stable_unclip.py +8 -6
- diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py +6 -6
- diffusers/pipelines/stable_diffusion_3/__init__.py +2 -0
- diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py +34 -3
- diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py +33 -7
- diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py +1201 -0
- diffusers/pipelines/stable_diffusion_attend_and_excite/pipeline_stable_diffusion_attend_and_excite.py +3 -3
- diffusers/pipelines/stable_diffusion_diffedit/pipeline_stable_diffusion_diffedit.py +6 -6
- diffusers/pipelines/stable_diffusion_gligen/pipeline_stable_diffusion_gligen.py +5 -5
- diffusers/pipelines/stable_diffusion_gligen/pipeline_stable_diffusion_gligen_text_image.py +5 -5
- diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_k_diffusion.py +6 -6
- diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py +0 -4
- diffusers/pipelines/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py +23 -29
- diffusers/pipelines/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py +27 -29
- diffusers/pipelines/stable_diffusion_sag/pipeline_stable_diffusion_sag.py +3 -3
- diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py +17 -27
- diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py +26 -29
- diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py +17 -145
- diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_instruct_pix2pix.py +0 -4
- diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_adapter.py +6 -6
- diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_xl_adapter.py +18 -28
- diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth.py +8 -6
- diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth_img2img.py +8 -6
- diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero.py +6 -4
- diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero_sdxl.py +0 -4
- diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py +3 -3
- diffusers/pipelines/wuerstchen/modeling_wuerstchen_prior.py +1 -1
- diffusers/pipelines/wuerstchen/pipeline_wuerstchen_prior.py +5 -4
- diffusers/schedulers/__init__.py +8 -0
- diffusers/schedulers/scheduling_cosine_dpmsolver_multistep.py +572 -0
- diffusers/schedulers/scheduling_ddim.py +1 -1
- diffusers/schedulers/scheduling_ddim_cogvideox.py +449 -0
- diffusers/schedulers/scheduling_ddpm.py +1 -1
- diffusers/schedulers/scheduling_ddpm_parallel.py +1 -1
- diffusers/schedulers/scheduling_deis_multistep.py +2 -2
- diffusers/schedulers/scheduling_dpm_cogvideox.py +489 -0
- diffusers/schedulers/scheduling_dpmsolver_multistep.py +1 -1
- diffusers/schedulers/scheduling_dpmsolver_multistep_inverse.py +1 -1
- diffusers/schedulers/scheduling_dpmsolver_singlestep.py +64 -19
- diffusers/schedulers/scheduling_edm_dpmsolver_multistep.py +2 -2
- diffusers/schedulers/scheduling_flow_match_euler_discrete.py +63 -39
- diffusers/schedulers/scheduling_flow_match_heun_discrete.py +321 -0
- diffusers/schedulers/scheduling_ipndm.py +1 -1
- diffusers/schedulers/scheduling_unipc_multistep.py +1 -1
- diffusers/schedulers/scheduling_utils.py +1 -3
- diffusers/schedulers/scheduling_utils_flax.py +1 -3
- diffusers/training_utils.py +99 -14
- diffusers/utils/__init__.py +2 -2
- diffusers/utils/dummy_pt_objects.py +210 -0
- diffusers/utils/dummy_torch_and_torchsde_objects.py +15 -0
- diffusers/utils/dummy_torch_and_transformers_and_sentencepiece_objects.py +47 -0
- diffusers/utils/dummy_torch_and_transformers_objects.py +315 -0
- diffusers/utils/dynamic_modules_utils.py +1 -11
- diffusers/utils/export_utils.py +50 -6
- diffusers/utils/hub_utils.py +45 -42
- diffusers/utils/import_utils.py +37 -15
- diffusers/utils/loading_utils.py +80 -3
- diffusers/utils/testing_utils.py +11 -8
- {diffusers-0.29.2.dist-info → diffusers-0.30.1.dist-info}/METADATA +73 -83
- {diffusers-0.29.2.dist-info → diffusers-0.30.1.dist-info}/RECORD +217 -164
- {diffusers-0.29.2.dist-info → diffusers-0.30.1.dist-info}/WHEEL +1 -1
- diffusers/loaders/autoencoder.py +0 -146
- diffusers/loaders/controlnet.py +0 -136
- diffusers/loaders/lora.py +0 -1728
- {diffusers-0.29.2.dist-info → diffusers-0.30.1.dist-info}/LICENSE +0 -0
- {diffusers-0.29.2.dist-info → diffusers-0.30.1.dist-info}/entry_points.txt +0 -0
- {diffusers-0.29.2.dist-info → diffusers-0.30.1.dist-info}/top_level.txt +0 -0
@@ -17,6 +17,21 @@ class AsymmetricAutoencoderKL(metaclass=DummyObject):
|
|
17
17
|
requires_backends(cls, ["torch"])
|
18
18
|
|
19
19
|
|
20
|
+
class AuraFlowTransformer2DModel(metaclass=DummyObject):
|
21
|
+
_backends = ["torch"]
|
22
|
+
|
23
|
+
def __init__(self, *args, **kwargs):
|
24
|
+
requires_backends(self, ["torch"])
|
25
|
+
|
26
|
+
@classmethod
|
27
|
+
def from_config(cls, *args, **kwargs):
|
28
|
+
requires_backends(cls, ["torch"])
|
29
|
+
|
30
|
+
@classmethod
|
31
|
+
def from_pretrained(cls, *args, **kwargs):
|
32
|
+
requires_backends(cls, ["torch"])
|
33
|
+
|
34
|
+
|
20
35
|
class AutoencoderKL(metaclass=DummyObject):
|
21
36
|
_backends = ["torch"]
|
22
37
|
|
@@ -32,6 +47,21 @@ class AutoencoderKL(metaclass=DummyObject):
|
|
32
47
|
requires_backends(cls, ["torch"])
|
33
48
|
|
34
49
|
|
50
|
+
class AutoencoderKLCogVideoX(metaclass=DummyObject):
|
51
|
+
_backends = ["torch"]
|
52
|
+
|
53
|
+
def __init__(self, *args, **kwargs):
|
54
|
+
requires_backends(self, ["torch"])
|
55
|
+
|
56
|
+
@classmethod
|
57
|
+
def from_config(cls, *args, **kwargs):
|
58
|
+
requires_backends(cls, ["torch"])
|
59
|
+
|
60
|
+
@classmethod
|
61
|
+
def from_pretrained(cls, *args, **kwargs):
|
62
|
+
requires_backends(cls, ["torch"])
|
63
|
+
|
64
|
+
|
35
65
|
class AutoencoderKLTemporalDecoder(metaclass=DummyObject):
|
36
66
|
_backends = ["torch"]
|
37
67
|
|
@@ -47,6 +77,21 @@ class AutoencoderKLTemporalDecoder(metaclass=DummyObject):
|
|
47
77
|
requires_backends(cls, ["torch"])
|
48
78
|
|
49
79
|
|
80
|
+
class AutoencoderOobleck(metaclass=DummyObject):
|
81
|
+
_backends = ["torch"]
|
82
|
+
|
83
|
+
def __init__(self, *args, **kwargs):
|
84
|
+
requires_backends(self, ["torch"])
|
85
|
+
|
86
|
+
@classmethod
|
87
|
+
def from_config(cls, *args, **kwargs):
|
88
|
+
requires_backends(cls, ["torch"])
|
89
|
+
|
90
|
+
@classmethod
|
91
|
+
def from_pretrained(cls, *args, **kwargs):
|
92
|
+
requires_backends(cls, ["torch"])
|
93
|
+
|
94
|
+
|
50
95
|
class AutoencoderTiny(metaclass=DummyObject):
|
51
96
|
_backends = ["torch"]
|
52
97
|
|
@@ -62,6 +107,21 @@ class AutoencoderTiny(metaclass=DummyObject):
|
|
62
107
|
requires_backends(cls, ["torch"])
|
63
108
|
|
64
109
|
|
110
|
+
class CogVideoXTransformer3DModel(metaclass=DummyObject):
|
111
|
+
_backends = ["torch"]
|
112
|
+
|
113
|
+
def __init__(self, *args, **kwargs):
|
114
|
+
requires_backends(self, ["torch"])
|
115
|
+
|
116
|
+
@classmethod
|
117
|
+
def from_config(cls, *args, **kwargs):
|
118
|
+
requires_backends(cls, ["torch"])
|
119
|
+
|
120
|
+
@classmethod
|
121
|
+
def from_pretrained(cls, *args, **kwargs):
|
122
|
+
requires_backends(cls, ["torch"])
|
123
|
+
|
124
|
+
|
65
125
|
class ConsistencyDecoderVAE(metaclass=DummyObject):
|
66
126
|
_backends = ["torch"]
|
67
127
|
|
@@ -122,6 +182,36 @@ class DiTTransformer2DModel(metaclass=DummyObject):
|
|
122
182
|
requires_backends(cls, ["torch"])
|
123
183
|
|
124
184
|
|
185
|
+
class FluxTransformer2DModel(metaclass=DummyObject):
|
186
|
+
_backends = ["torch"]
|
187
|
+
|
188
|
+
def __init__(self, *args, **kwargs):
|
189
|
+
requires_backends(self, ["torch"])
|
190
|
+
|
191
|
+
@classmethod
|
192
|
+
def from_config(cls, *args, **kwargs):
|
193
|
+
requires_backends(cls, ["torch"])
|
194
|
+
|
195
|
+
@classmethod
|
196
|
+
def from_pretrained(cls, *args, **kwargs):
|
197
|
+
requires_backends(cls, ["torch"])
|
198
|
+
|
199
|
+
|
200
|
+
class HunyuanDiT2DControlNetModel(metaclass=DummyObject):
|
201
|
+
_backends = ["torch"]
|
202
|
+
|
203
|
+
def __init__(self, *args, **kwargs):
|
204
|
+
requires_backends(self, ["torch"])
|
205
|
+
|
206
|
+
@classmethod
|
207
|
+
def from_config(cls, *args, **kwargs):
|
208
|
+
requires_backends(cls, ["torch"])
|
209
|
+
|
210
|
+
@classmethod
|
211
|
+
def from_pretrained(cls, *args, **kwargs):
|
212
|
+
requires_backends(cls, ["torch"])
|
213
|
+
|
214
|
+
|
125
215
|
class HunyuanDiT2DModel(metaclass=DummyObject):
|
126
216
|
_backends = ["torch"]
|
127
217
|
|
@@ -137,6 +227,21 @@ class HunyuanDiT2DModel(metaclass=DummyObject):
|
|
137
227
|
requires_backends(cls, ["torch"])
|
138
228
|
|
139
229
|
|
230
|
+
class HunyuanDiT2DMultiControlNetModel(metaclass=DummyObject):
|
231
|
+
_backends = ["torch"]
|
232
|
+
|
233
|
+
def __init__(self, *args, **kwargs):
|
234
|
+
requires_backends(self, ["torch"])
|
235
|
+
|
236
|
+
@classmethod
|
237
|
+
def from_config(cls, *args, **kwargs):
|
238
|
+
requires_backends(cls, ["torch"])
|
239
|
+
|
240
|
+
@classmethod
|
241
|
+
def from_pretrained(cls, *args, **kwargs):
|
242
|
+
requires_backends(cls, ["torch"])
|
243
|
+
|
244
|
+
|
140
245
|
class I2VGenXLUNet(metaclass=DummyObject):
|
141
246
|
_backends = ["torch"]
|
142
247
|
|
@@ -167,6 +272,36 @@ class Kandinsky3UNet(metaclass=DummyObject):
|
|
167
272
|
requires_backends(cls, ["torch"])
|
168
273
|
|
169
274
|
|
275
|
+
class LatteTransformer3DModel(metaclass=DummyObject):
|
276
|
+
_backends = ["torch"]
|
277
|
+
|
278
|
+
def __init__(self, *args, **kwargs):
|
279
|
+
requires_backends(self, ["torch"])
|
280
|
+
|
281
|
+
@classmethod
|
282
|
+
def from_config(cls, *args, **kwargs):
|
283
|
+
requires_backends(cls, ["torch"])
|
284
|
+
|
285
|
+
@classmethod
|
286
|
+
def from_pretrained(cls, *args, **kwargs):
|
287
|
+
requires_backends(cls, ["torch"])
|
288
|
+
|
289
|
+
|
290
|
+
class LuminaNextDiT2DModel(metaclass=DummyObject):
|
291
|
+
_backends = ["torch"]
|
292
|
+
|
293
|
+
def __init__(self, *args, **kwargs):
|
294
|
+
requires_backends(self, ["torch"])
|
295
|
+
|
296
|
+
@classmethod
|
297
|
+
def from_config(cls, *args, **kwargs):
|
298
|
+
requires_backends(cls, ["torch"])
|
299
|
+
|
300
|
+
@classmethod
|
301
|
+
def from_pretrained(cls, *args, **kwargs):
|
302
|
+
requires_backends(cls, ["torch"])
|
303
|
+
|
304
|
+
|
170
305
|
class ModelMixin(metaclass=DummyObject):
|
171
306
|
_backends = ["torch"]
|
172
307
|
|
@@ -287,6 +422,36 @@ class SD3Transformer2DModel(metaclass=DummyObject):
|
|
287
422
|
requires_backends(cls, ["torch"])
|
288
423
|
|
289
424
|
|
425
|
+
class SparseControlNetModel(metaclass=DummyObject):
|
426
|
+
_backends = ["torch"]
|
427
|
+
|
428
|
+
def __init__(self, *args, **kwargs):
|
429
|
+
requires_backends(self, ["torch"])
|
430
|
+
|
431
|
+
@classmethod
|
432
|
+
def from_config(cls, *args, **kwargs):
|
433
|
+
requires_backends(cls, ["torch"])
|
434
|
+
|
435
|
+
@classmethod
|
436
|
+
def from_pretrained(cls, *args, **kwargs):
|
437
|
+
requires_backends(cls, ["torch"])
|
438
|
+
|
439
|
+
|
440
|
+
class StableAudioDiTModel(metaclass=DummyObject):
|
441
|
+
_backends = ["torch"]
|
442
|
+
|
443
|
+
def __init__(self, *args, **kwargs):
|
444
|
+
requires_backends(self, ["torch"])
|
445
|
+
|
446
|
+
@classmethod
|
447
|
+
def from_config(cls, *args, **kwargs):
|
448
|
+
requires_backends(cls, ["torch"])
|
449
|
+
|
450
|
+
@classmethod
|
451
|
+
def from_pretrained(cls, *args, **kwargs):
|
452
|
+
requires_backends(cls, ["torch"])
|
453
|
+
|
454
|
+
|
290
455
|
class T2IAdapter(metaclass=DummyObject):
|
291
456
|
_backends = ["torch"]
|
292
457
|
|
@@ -840,6 +1005,36 @@ class CMStochasticIterativeScheduler(metaclass=DummyObject):
|
|
840
1005
|
requires_backends(cls, ["torch"])
|
841
1006
|
|
842
1007
|
|
1008
|
+
class CogVideoXDDIMScheduler(metaclass=DummyObject):
|
1009
|
+
_backends = ["torch"]
|
1010
|
+
|
1011
|
+
def __init__(self, *args, **kwargs):
|
1012
|
+
requires_backends(self, ["torch"])
|
1013
|
+
|
1014
|
+
@classmethod
|
1015
|
+
def from_config(cls, *args, **kwargs):
|
1016
|
+
requires_backends(cls, ["torch"])
|
1017
|
+
|
1018
|
+
@classmethod
|
1019
|
+
def from_pretrained(cls, *args, **kwargs):
|
1020
|
+
requires_backends(cls, ["torch"])
|
1021
|
+
|
1022
|
+
|
1023
|
+
class CogVideoXDPMScheduler(metaclass=DummyObject):
|
1024
|
+
_backends = ["torch"]
|
1025
|
+
|
1026
|
+
def __init__(self, *args, **kwargs):
|
1027
|
+
requires_backends(self, ["torch"])
|
1028
|
+
|
1029
|
+
@classmethod
|
1030
|
+
def from_config(cls, *args, **kwargs):
|
1031
|
+
requires_backends(cls, ["torch"])
|
1032
|
+
|
1033
|
+
@classmethod
|
1034
|
+
def from_pretrained(cls, *args, **kwargs):
|
1035
|
+
requires_backends(cls, ["torch"])
|
1036
|
+
|
1037
|
+
|
843
1038
|
class DDIMInverseScheduler(metaclass=DummyObject):
|
844
1039
|
_backends = ["torch"]
|
845
1040
|
|
@@ -1065,6 +1260,21 @@ class FlowMatchEulerDiscreteScheduler(metaclass=DummyObject):
|
|
1065
1260
|
requires_backends(cls, ["torch"])
|
1066
1261
|
|
1067
1262
|
|
1263
|
+
class FlowMatchHeunDiscreteScheduler(metaclass=DummyObject):
|
1264
|
+
_backends = ["torch"]
|
1265
|
+
|
1266
|
+
def __init__(self, *args, **kwargs):
|
1267
|
+
requires_backends(self, ["torch"])
|
1268
|
+
|
1269
|
+
@classmethod
|
1270
|
+
def from_config(cls, *args, **kwargs):
|
1271
|
+
requires_backends(cls, ["torch"])
|
1272
|
+
|
1273
|
+
@classmethod
|
1274
|
+
def from_pretrained(cls, *args, **kwargs):
|
1275
|
+
requires_backends(cls, ["torch"])
|
1276
|
+
|
1277
|
+
|
1068
1278
|
class HeunDiscreteScheduler(metaclass=DummyObject):
|
1069
1279
|
_backends = ["torch"]
|
1070
1280
|
|
@@ -2,6 +2,21 @@
|
|
2
2
|
from ..utils import DummyObject, requires_backends
|
3
3
|
|
4
4
|
|
5
|
+
class CosineDPMSolverMultistepScheduler(metaclass=DummyObject):
|
6
|
+
_backends = ["torch", "torchsde"]
|
7
|
+
|
8
|
+
def __init__(self, *args, **kwargs):
|
9
|
+
requires_backends(self, ["torch", "torchsde"])
|
10
|
+
|
11
|
+
@classmethod
|
12
|
+
def from_config(cls, *args, **kwargs):
|
13
|
+
requires_backends(cls, ["torch", "torchsde"])
|
14
|
+
|
15
|
+
@classmethod
|
16
|
+
def from_pretrained(cls, *args, **kwargs):
|
17
|
+
requires_backends(cls, ["torch", "torchsde"])
|
18
|
+
|
19
|
+
|
5
20
|
class DPMSolverSDEScheduler(metaclass=DummyObject):
|
6
21
|
_backends = ["torch", "torchsde"]
|
7
22
|
|
@@ -0,0 +1,47 @@
|
|
1
|
+
# This file is autogenerated by the command `make fix-copies`, do not edit.
|
2
|
+
from ..utils import DummyObject, requires_backends
|
3
|
+
|
4
|
+
|
5
|
+
class KolorsImg2ImgPipeline(metaclass=DummyObject):
|
6
|
+
_backends = ["torch", "transformers", "sentencepiece"]
|
7
|
+
|
8
|
+
def __init__(self, *args, **kwargs):
|
9
|
+
requires_backends(self, ["torch", "transformers", "sentencepiece"])
|
10
|
+
|
11
|
+
@classmethod
|
12
|
+
def from_config(cls, *args, **kwargs):
|
13
|
+
requires_backends(cls, ["torch", "transformers", "sentencepiece"])
|
14
|
+
|
15
|
+
@classmethod
|
16
|
+
def from_pretrained(cls, *args, **kwargs):
|
17
|
+
requires_backends(cls, ["torch", "transformers", "sentencepiece"])
|
18
|
+
|
19
|
+
|
20
|
+
class KolorsPAGPipeline(metaclass=DummyObject):
|
21
|
+
_backends = ["torch", "transformers", "sentencepiece"]
|
22
|
+
|
23
|
+
def __init__(self, *args, **kwargs):
|
24
|
+
requires_backends(self, ["torch", "transformers", "sentencepiece"])
|
25
|
+
|
26
|
+
@classmethod
|
27
|
+
def from_config(cls, *args, **kwargs):
|
28
|
+
requires_backends(cls, ["torch", "transformers", "sentencepiece"])
|
29
|
+
|
30
|
+
@classmethod
|
31
|
+
def from_pretrained(cls, *args, **kwargs):
|
32
|
+
requires_backends(cls, ["torch", "transformers", "sentencepiece"])
|
33
|
+
|
34
|
+
|
35
|
+
class KolorsPipeline(metaclass=DummyObject):
|
36
|
+
_backends = ["torch", "transformers", "sentencepiece"]
|
37
|
+
|
38
|
+
def __init__(self, *args, **kwargs):
|
39
|
+
requires_backends(self, ["torch", "transformers", "sentencepiece"])
|
40
|
+
|
41
|
+
@classmethod
|
42
|
+
def from_config(cls, *args, **kwargs):
|
43
|
+
requires_backends(cls, ["torch", "transformers", "sentencepiece"])
|
44
|
+
|
45
|
+
@classmethod
|
46
|
+
def from_pretrained(cls, *args, **kwargs):
|
47
|
+
requires_backends(cls, ["torch", "transformers", "sentencepiece"])
|