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.
Files changed (220) hide show
  1. diffusers/__init__.py +94 -3
  2. diffusers/commands/env.py +1 -5
  3. diffusers/configuration_utils.py +4 -9
  4. diffusers/dependency_versions_table.py +2 -2
  5. diffusers/image_processor.py +1 -2
  6. diffusers/loaders/__init__.py +17 -2
  7. diffusers/loaders/ip_adapter.py +10 -7
  8. diffusers/loaders/lora_base.py +752 -0
  9. diffusers/loaders/lora_pipeline.py +2252 -0
  10. diffusers/loaders/peft.py +213 -5
  11. diffusers/loaders/single_file.py +3 -14
  12. diffusers/loaders/single_file_model.py +31 -10
  13. diffusers/loaders/single_file_utils.py +293 -8
  14. diffusers/loaders/textual_inversion.py +1 -6
  15. diffusers/loaders/unet.py +23 -208
  16. diffusers/models/__init__.py +20 -0
  17. diffusers/models/activations.py +22 -0
  18. diffusers/models/attention.py +386 -7
  19. diffusers/models/attention_processor.py +1937 -629
  20. diffusers/models/autoencoders/__init__.py +2 -0
  21. diffusers/models/autoencoders/autoencoder_kl.py +14 -3
  22. diffusers/models/autoencoders/autoencoder_kl_cogvideox.py +1271 -0
  23. diffusers/models/autoencoders/autoencoder_kl_temporal_decoder.py +1 -1
  24. diffusers/models/autoencoders/autoencoder_oobleck.py +464 -0
  25. diffusers/models/autoencoders/autoencoder_tiny.py +1 -0
  26. diffusers/models/autoencoders/consistency_decoder_vae.py +1 -1
  27. diffusers/models/autoencoders/vq_model.py +4 -4
  28. diffusers/models/controlnet.py +2 -3
  29. diffusers/models/controlnet_hunyuan.py +401 -0
  30. diffusers/models/controlnet_sd3.py +11 -11
  31. diffusers/models/controlnet_sparsectrl.py +789 -0
  32. diffusers/models/controlnet_xs.py +40 -10
  33. diffusers/models/downsampling.py +68 -0
  34. diffusers/models/embeddings.py +403 -36
  35. diffusers/models/model_loading_utils.py +1 -3
  36. diffusers/models/modeling_flax_utils.py +1 -6
  37. diffusers/models/modeling_utils.py +4 -16
  38. diffusers/models/normalization.py +203 -12
  39. diffusers/models/transformers/__init__.py +6 -0
  40. diffusers/models/transformers/auraflow_transformer_2d.py +543 -0
  41. diffusers/models/transformers/cogvideox_transformer_3d.py +485 -0
  42. diffusers/models/transformers/hunyuan_transformer_2d.py +19 -15
  43. diffusers/models/transformers/latte_transformer_3d.py +327 -0
  44. diffusers/models/transformers/lumina_nextdit2d.py +340 -0
  45. diffusers/models/transformers/pixart_transformer_2d.py +102 -1
  46. diffusers/models/transformers/prior_transformer.py +1 -1
  47. diffusers/models/transformers/stable_audio_transformer.py +458 -0
  48. diffusers/models/transformers/transformer_flux.py +455 -0
  49. diffusers/models/transformers/transformer_sd3.py +18 -4
  50. diffusers/models/unets/unet_1d_blocks.py +1 -1
  51. diffusers/models/unets/unet_2d_condition.py +8 -1
  52. diffusers/models/unets/unet_3d_blocks.py +51 -920
  53. diffusers/models/unets/unet_3d_condition.py +4 -1
  54. diffusers/models/unets/unet_i2vgen_xl.py +4 -1
  55. diffusers/models/unets/unet_kandinsky3.py +1 -1
  56. diffusers/models/unets/unet_motion_model.py +1330 -84
  57. diffusers/models/unets/unet_spatio_temporal_condition.py +1 -1
  58. diffusers/models/unets/unet_stable_cascade.py +1 -3
  59. diffusers/models/unets/uvit_2d.py +1 -1
  60. diffusers/models/upsampling.py +64 -0
  61. diffusers/models/vq_model.py +8 -4
  62. diffusers/optimization.py +1 -1
  63. diffusers/pipelines/__init__.py +100 -3
  64. diffusers/pipelines/animatediff/__init__.py +4 -0
  65. diffusers/pipelines/animatediff/pipeline_animatediff.py +50 -40
  66. diffusers/pipelines/animatediff/pipeline_animatediff_controlnet.py +1076 -0
  67. diffusers/pipelines/animatediff/pipeline_animatediff_sdxl.py +17 -27
  68. diffusers/pipelines/animatediff/pipeline_animatediff_sparsectrl.py +1008 -0
  69. diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py +51 -38
  70. diffusers/pipelines/audioldm2/modeling_audioldm2.py +1 -1
  71. diffusers/pipelines/audioldm2/pipeline_audioldm2.py +1 -0
  72. diffusers/pipelines/aura_flow/__init__.py +48 -0
  73. diffusers/pipelines/aura_flow/pipeline_aura_flow.py +591 -0
  74. diffusers/pipelines/auto_pipeline.py +97 -19
  75. diffusers/pipelines/cogvideo/__init__.py +48 -0
  76. diffusers/pipelines/cogvideo/pipeline_cogvideox.py +746 -0
  77. diffusers/pipelines/consistency_models/pipeline_consistency_models.py +1 -1
  78. diffusers/pipelines/controlnet/pipeline_controlnet.py +24 -30
  79. diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py +31 -30
  80. diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py +24 -153
  81. diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py +19 -28
  82. diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py +18 -28
  83. diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py +29 -32
  84. diffusers/pipelines/controlnet/pipeline_flax_controlnet.py +2 -2
  85. diffusers/pipelines/controlnet_hunyuandit/__init__.py +48 -0
  86. diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py +1042 -0
  87. diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet.py +35 -0
  88. diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs.py +10 -6
  89. diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs_sd_xl.py +0 -4
  90. diffusers/pipelines/deepfloyd_if/pipeline_if.py +2 -2
  91. diffusers/pipelines/deepfloyd_if/pipeline_if_img2img.py +2 -2
  92. diffusers/pipelines/deepfloyd_if/pipeline_if_img2img_superresolution.py +2 -2
  93. diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting.py +2 -2
  94. diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting_superresolution.py +2 -2
  95. diffusers/pipelines/deepfloyd_if/pipeline_if_superresolution.py +2 -2
  96. diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py +11 -6
  97. diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py +11 -6
  98. diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_cycle_diffusion.py +6 -6
  99. diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_inpaint_legacy.py +6 -6
  100. diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_model_editing.py +10 -10
  101. diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_paradigms.py +10 -6
  102. diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_pix2pix_zero.py +3 -3
  103. diffusers/pipelines/deprecated/versatile_diffusion/modeling_text_unet.py +1 -1
  104. diffusers/pipelines/flux/__init__.py +47 -0
  105. diffusers/pipelines/flux/pipeline_flux.py +749 -0
  106. diffusers/pipelines/flux/pipeline_output.py +21 -0
  107. diffusers/pipelines/free_init_utils.py +2 -0
  108. diffusers/pipelines/free_noise_utils.py +236 -0
  109. diffusers/pipelines/kandinsky3/pipeline_kandinsky3.py +2 -2
  110. diffusers/pipelines/kandinsky3/pipeline_kandinsky3_img2img.py +2 -2
  111. diffusers/pipelines/kolors/__init__.py +54 -0
  112. diffusers/pipelines/kolors/pipeline_kolors.py +1070 -0
  113. diffusers/pipelines/kolors/pipeline_kolors_img2img.py +1247 -0
  114. diffusers/pipelines/kolors/pipeline_output.py +21 -0
  115. diffusers/pipelines/kolors/text_encoder.py +889 -0
  116. diffusers/pipelines/kolors/tokenizer.py +334 -0
  117. diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_img2img.py +30 -29
  118. diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_text2img.py +23 -29
  119. diffusers/pipelines/latte/__init__.py +48 -0
  120. diffusers/pipelines/latte/pipeline_latte.py +881 -0
  121. diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py +4 -4
  122. diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py +0 -4
  123. diffusers/pipelines/lumina/__init__.py +48 -0
  124. diffusers/pipelines/lumina/pipeline_lumina.py +897 -0
  125. diffusers/pipelines/pag/__init__.py +67 -0
  126. diffusers/pipelines/pag/pag_utils.py +237 -0
  127. diffusers/pipelines/pag/pipeline_pag_controlnet_sd.py +1329 -0
  128. diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl.py +1612 -0
  129. diffusers/pipelines/pag/pipeline_pag_hunyuandit.py +953 -0
  130. diffusers/pipelines/pag/pipeline_pag_kolors.py +1136 -0
  131. diffusers/pipelines/pag/pipeline_pag_pixart_sigma.py +872 -0
  132. diffusers/pipelines/pag/pipeline_pag_sd.py +1050 -0
  133. diffusers/pipelines/pag/pipeline_pag_sd_3.py +985 -0
  134. diffusers/pipelines/pag/pipeline_pag_sd_animatediff.py +862 -0
  135. diffusers/pipelines/pag/pipeline_pag_sd_xl.py +1333 -0
  136. diffusers/pipelines/pag/pipeline_pag_sd_xl_img2img.py +1529 -0
  137. diffusers/pipelines/pag/pipeline_pag_sd_xl_inpaint.py +1753 -0
  138. diffusers/pipelines/pia/pipeline_pia.py +30 -37
  139. diffusers/pipelines/pipeline_flax_utils.py +4 -9
  140. diffusers/pipelines/pipeline_loading_utils.py +0 -3
  141. diffusers/pipelines/pipeline_utils.py +2 -14
  142. diffusers/pipelines/semantic_stable_diffusion/pipeline_semantic_stable_diffusion.py +0 -1
  143. diffusers/pipelines/stable_audio/__init__.py +50 -0
  144. diffusers/pipelines/stable_audio/modeling_stable_audio.py +158 -0
  145. diffusers/pipelines/stable_audio/pipeline_stable_audio.py +745 -0
  146. diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +2 -0
  147. diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion.py +1 -1
  148. diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py +23 -29
  149. diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py +15 -8
  150. diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py +30 -29
  151. diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py +23 -152
  152. diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py +8 -4
  153. diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py +11 -11
  154. diffusers/pipelines/stable_diffusion/pipeline_stable_unclip.py +8 -6
  155. diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py +6 -6
  156. diffusers/pipelines/stable_diffusion_3/__init__.py +2 -0
  157. diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py +34 -3
  158. diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py +33 -7
  159. diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py +1201 -0
  160. diffusers/pipelines/stable_diffusion_attend_and_excite/pipeline_stable_diffusion_attend_and_excite.py +3 -3
  161. diffusers/pipelines/stable_diffusion_diffedit/pipeline_stable_diffusion_diffedit.py +6 -6
  162. diffusers/pipelines/stable_diffusion_gligen/pipeline_stable_diffusion_gligen.py +5 -5
  163. diffusers/pipelines/stable_diffusion_gligen/pipeline_stable_diffusion_gligen_text_image.py +5 -5
  164. diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_k_diffusion.py +6 -6
  165. diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py +0 -4
  166. diffusers/pipelines/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py +23 -29
  167. diffusers/pipelines/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py +27 -29
  168. diffusers/pipelines/stable_diffusion_sag/pipeline_stable_diffusion_sag.py +3 -3
  169. diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py +17 -27
  170. diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py +26 -29
  171. diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py +17 -145
  172. diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_instruct_pix2pix.py +0 -4
  173. diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_adapter.py +6 -6
  174. diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_xl_adapter.py +18 -28
  175. diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth.py +8 -6
  176. diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth_img2img.py +8 -6
  177. diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero.py +6 -4
  178. diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero_sdxl.py +0 -4
  179. diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py +3 -3
  180. diffusers/pipelines/wuerstchen/modeling_wuerstchen_prior.py +1 -1
  181. diffusers/pipelines/wuerstchen/pipeline_wuerstchen_prior.py +5 -4
  182. diffusers/schedulers/__init__.py +8 -0
  183. diffusers/schedulers/scheduling_cosine_dpmsolver_multistep.py +572 -0
  184. diffusers/schedulers/scheduling_ddim.py +1 -1
  185. diffusers/schedulers/scheduling_ddim_cogvideox.py +449 -0
  186. diffusers/schedulers/scheduling_ddpm.py +1 -1
  187. diffusers/schedulers/scheduling_ddpm_parallel.py +1 -1
  188. diffusers/schedulers/scheduling_deis_multistep.py +2 -2
  189. diffusers/schedulers/scheduling_dpm_cogvideox.py +489 -0
  190. diffusers/schedulers/scheduling_dpmsolver_multistep.py +1 -1
  191. diffusers/schedulers/scheduling_dpmsolver_multistep_inverse.py +1 -1
  192. diffusers/schedulers/scheduling_dpmsolver_singlestep.py +64 -19
  193. diffusers/schedulers/scheduling_edm_dpmsolver_multistep.py +2 -2
  194. diffusers/schedulers/scheduling_flow_match_euler_discrete.py +63 -39
  195. diffusers/schedulers/scheduling_flow_match_heun_discrete.py +321 -0
  196. diffusers/schedulers/scheduling_ipndm.py +1 -1
  197. diffusers/schedulers/scheduling_unipc_multistep.py +1 -1
  198. diffusers/schedulers/scheduling_utils.py +1 -3
  199. diffusers/schedulers/scheduling_utils_flax.py +1 -3
  200. diffusers/training_utils.py +99 -14
  201. diffusers/utils/__init__.py +2 -2
  202. diffusers/utils/dummy_pt_objects.py +210 -0
  203. diffusers/utils/dummy_torch_and_torchsde_objects.py +15 -0
  204. diffusers/utils/dummy_torch_and_transformers_and_sentencepiece_objects.py +47 -0
  205. diffusers/utils/dummy_torch_and_transformers_objects.py +315 -0
  206. diffusers/utils/dynamic_modules_utils.py +1 -11
  207. diffusers/utils/export_utils.py +50 -6
  208. diffusers/utils/hub_utils.py +45 -42
  209. diffusers/utils/import_utils.py +37 -15
  210. diffusers/utils/loading_utils.py +80 -3
  211. diffusers/utils/testing_utils.py +11 -8
  212. {diffusers-0.29.2.dist-info → diffusers-0.30.1.dist-info}/METADATA +73 -83
  213. {diffusers-0.29.2.dist-info → diffusers-0.30.1.dist-info}/RECORD +217 -164
  214. {diffusers-0.29.2.dist-info → diffusers-0.30.1.dist-info}/WHEEL +1 -1
  215. diffusers/loaders/autoencoder.py +0 -146
  216. diffusers/loaders/controlnet.py +0 -136
  217. diffusers/loaders/lora.py +0 -1728
  218. {diffusers-0.29.2.dist-info → diffusers-0.30.1.dist-info}/LICENSE +0 -0
  219. {diffusers-0.29.2.dist-info → diffusers-0.30.1.dist-info}/entry_points.txt +0 -0
  220. {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"])