diffusers 0.27.0__py3-none-any.whl → 0.32.2__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (445) hide show
  1. diffusers/__init__.py +233 -6
  2. diffusers/callbacks.py +209 -0
  3. diffusers/commands/env.py +102 -6
  4. diffusers/configuration_utils.py +45 -16
  5. diffusers/dependency_versions_table.py +4 -3
  6. diffusers/image_processor.py +434 -110
  7. diffusers/loaders/__init__.py +42 -9
  8. diffusers/loaders/ip_adapter.py +626 -36
  9. diffusers/loaders/lora_base.py +900 -0
  10. diffusers/loaders/lora_conversion_utils.py +991 -125
  11. diffusers/loaders/lora_pipeline.py +3812 -0
  12. diffusers/loaders/peft.py +571 -7
  13. diffusers/loaders/single_file.py +405 -173
  14. diffusers/loaders/single_file_model.py +385 -0
  15. diffusers/loaders/single_file_utils.py +1783 -713
  16. diffusers/loaders/textual_inversion.py +41 -23
  17. diffusers/loaders/transformer_flux.py +181 -0
  18. diffusers/loaders/transformer_sd3.py +89 -0
  19. diffusers/loaders/unet.py +464 -540
  20. diffusers/loaders/unet_loader_utils.py +163 -0
  21. diffusers/models/__init__.py +76 -7
  22. diffusers/models/activations.py +65 -10
  23. diffusers/models/adapter.py +53 -53
  24. diffusers/models/attention.py +605 -18
  25. diffusers/models/attention_flax.py +1 -1
  26. diffusers/models/attention_processor.py +4304 -687
  27. diffusers/models/autoencoders/__init__.py +8 -0
  28. diffusers/models/autoencoders/autoencoder_asym_kl.py +15 -17
  29. diffusers/models/autoencoders/autoencoder_dc.py +620 -0
  30. diffusers/models/autoencoders/autoencoder_kl.py +110 -28
  31. diffusers/models/autoencoders/autoencoder_kl_allegro.py +1149 -0
  32. diffusers/models/autoencoders/autoencoder_kl_cogvideox.py +1482 -0
  33. diffusers/models/autoencoders/autoencoder_kl_hunyuan_video.py +1176 -0
  34. diffusers/models/autoencoders/autoencoder_kl_ltx.py +1338 -0
  35. diffusers/models/autoencoders/autoencoder_kl_mochi.py +1166 -0
  36. diffusers/models/autoencoders/autoencoder_kl_temporal_decoder.py +19 -24
  37. diffusers/models/autoencoders/autoencoder_oobleck.py +464 -0
  38. diffusers/models/autoencoders/autoencoder_tiny.py +21 -18
  39. diffusers/models/autoencoders/consistency_decoder_vae.py +45 -20
  40. diffusers/models/autoencoders/vae.py +41 -29
  41. diffusers/models/autoencoders/vq_model.py +182 -0
  42. diffusers/models/controlnet.py +47 -800
  43. diffusers/models/controlnet_flux.py +70 -0
  44. diffusers/models/controlnet_sd3.py +68 -0
  45. diffusers/models/controlnet_sparsectrl.py +116 -0
  46. diffusers/models/controlnets/__init__.py +23 -0
  47. diffusers/models/controlnets/controlnet.py +872 -0
  48. diffusers/models/{controlnet_flax.py → controlnets/controlnet_flax.py} +9 -9
  49. diffusers/models/controlnets/controlnet_flux.py +536 -0
  50. diffusers/models/controlnets/controlnet_hunyuan.py +401 -0
  51. diffusers/models/controlnets/controlnet_sd3.py +489 -0
  52. diffusers/models/controlnets/controlnet_sparsectrl.py +788 -0
  53. diffusers/models/controlnets/controlnet_union.py +832 -0
  54. diffusers/models/controlnets/controlnet_xs.py +1946 -0
  55. diffusers/models/controlnets/multicontrolnet.py +183 -0
  56. diffusers/models/downsampling.py +85 -18
  57. diffusers/models/embeddings.py +1856 -158
  58. diffusers/models/embeddings_flax.py +23 -9
  59. diffusers/models/model_loading_utils.py +480 -0
  60. diffusers/models/modeling_flax_pytorch_utils.py +2 -1
  61. diffusers/models/modeling_flax_utils.py +2 -7
  62. diffusers/models/modeling_outputs.py +14 -0
  63. diffusers/models/modeling_pytorch_flax_utils.py +1 -1
  64. diffusers/models/modeling_utils.py +611 -146
  65. diffusers/models/normalization.py +361 -20
  66. diffusers/models/resnet.py +18 -23
  67. diffusers/models/transformers/__init__.py +16 -0
  68. diffusers/models/transformers/auraflow_transformer_2d.py +544 -0
  69. diffusers/models/transformers/cogvideox_transformer_3d.py +542 -0
  70. diffusers/models/transformers/dit_transformer_2d.py +240 -0
  71. diffusers/models/transformers/dual_transformer_2d.py +9 -8
  72. diffusers/models/transformers/hunyuan_transformer_2d.py +578 -0
  73. diffusers/models/transformers/latte_transformer_3d.py +327 -0
  74. diffusers/models/transformers/lumina_nextdit2d.py +340 -0
  75. diffusers/models/transformers/pixart_transformer_2d.py +445 -0
  76. diffusers/models/transformers/prior_transformer.py +13 -13
  77. diffusers/models/transformers/sana_transformer.py +488 -0
  78. diffusers/models/transformers/stable_audio_transformer.py +458 -0
  79. diffusers/models/transformers/t5_film_transformer.py +17 -19
  80. diffusers/models/transformers/transformer_2d.py +297 -187
  81. diffusers/models/transformers/transformer_allegro.py +422 -0
  82. diffusers/models/transformers/transformer_cogview3plus.py +386 -0
  83. diffusers/models/transformers/transformer_flux.py +593 -0
  84. diffusers/models/transformers/transformer_hunyuan_video.py +791 -0
  85. diffusers/models/transformers/transformer_ltx.py +469 -0
  86. diffusers/models/transformers/transformer_mochi.py +499 -0
  87. diffusers/models/transformers/transformer_sd3.py +461 -0
  88. diffusers/models/transformers/transformer_temporal.py +21 -19
  89. diffusers/models/unets/unet_1d.py +8 -8
  90. diffusers/models/unets/unet_1d_blocks.py +31 -31
  91. diffusers/models/unets/unet_2d.py +17 -10
  92. diffusers/models/unets/unet_2d_blocks.py +225 -149
  93. diffusers/models/unets/unet_2d_condition.py +50 -53
  94. diffusers/models/unets/unet_2d_condition_flax.py +6 -5
  95. diffusers/models/unets/unet_3d_blocks.py +192 -1057
  96. diffusers/models/unets/unet_3d_condition.py +22 -27
  97. diffusers/models/unets/unet_i2vgen_xl.py +22 -18
  98. diffusers/models/unets/unet_kandinsky3.py +2 -2
  99. diffusers/models/unets/unet_motion_model.py +1413 -89
  100. diffusers/models/unets/unet_spatio_temporal_condition.py +40 -16
  101. diffusers/models/unets/unet_stable_cascade.py +19 -18
  102. diffusers/models/unets/uvit_2d.py +2 -2
  103. diffusers/models/upsampling.py +95 -26
  104. diffusers/models/vq_model.py +12 -164
  105. diffusers/optimization.py +1 -1
  106. diffusers/pipelines/__init__.py +202 -3
  107. diffusers/pipelines/allegro/__init__.py +48 -0
  108. diffusers/pipelines/allegro/pipeline_allegro.py +938 -0
  109. diffusers/pipelines/allegro/pipeline_output.py +23 -0
  110. diffusers/pipelines/amused/pipeline_amused.py +12 -12
  111. diffusers/pipelines/amused/pipeline_amused_img2img.py +14 -12
  112. diffusers/pipelines/amused/pipeline_amused_inpaint.py +13 -11
  113. diffusers/pipelines/animatediff/__init__.py +8 -0
  114. diffusers/pipelines/animatediff/pipeline_animatediff.py +122 -109
  115. diffusers/pipelines/animatediff/pipeline_animatediff_controlnet.py +1106 -0
  116. diffusers/pipelines/animatediff/pipeline_animatediff_sdxl.py +1288 -0
  117. diffusers/pipelines/animatediff/pipeline_animatediff_sparsectrl.py +1010 -0
  118. diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py +236 -180
  119. diffusers/pipelines/animatediff/pipeline_animatediff_video2video_controlnet.py +1341 -0
  120. diffusers/pipelines/animatediff/pipeline_output.py +3 -2
  121. diffusers/pipelines/audioldm/pipeline_audioldm.py +14 -14
  122. diffusers/pipelines/audioldm2/modeling_audioldm2.py +58 -39
  123. diffusers/pipelines/audioldm2/pipeline_audioldm2.py +121 -36
  124. diffusers/pipelines/aura_flow/__init__.py +48 -0
  125. diffusers/pipelines/aura_flow/pipeline_aura_flow.py +584 -0
  126. diffusers/pipelines/auto_pipeline.py +196 -28
  127. diffusers/pipelines/blip_diffusion/blip_image_processing.py +1 -1
  128. diffusers/pipelines/blip_diffusion/modeling_blip2.py +6 -6
  129. diffusers/pipelines/blip_diffusion/modeling_ctx_clip.py +1 -1
  130. diffusers/pipelines/blip_diffusion/pipeline_blip_diffusion.py +2 -2
  131. diffusers/pipelines/cogvideo/__init__.py +54 -0
  132. diffusers/pipelines/cogvideo/pipeline_cogvideox.py +772 -0
  133. diffusers/pipelines/cogvideo/pipeline_cogvideox_fun_control.py +825 -0
  134. diffusers/pipelines/cogvideo/pipeline_cogvideox_image2video.py +885 -0
  135. diffusers/pipelines/cogvideo/pipeline_cogvideox_video2video.py +851 -0
  136. diffusers/pipelines/cogvideo/pipeline_output.py +20 -0
  137. diffusers/pipelines/cogview3/__init__.py +47 -0
  138. diffusers/pipelines/cogview3/pipeline_cogview3plus.py +674 -0
  139. diffusers/pipelines/cogview3/pipeline_output.py +21 -0
  140. diffusers/pipelines/consistency_models/pipeline_consistency_models.py +6 -6
  141. diffusers/pipelines/controlnet/__init__.py +86 -80
  142. diffusers/pipelines/controlnet/multicontrolnet.py +7 -182
  143. diffusers/pipelines/controlnet/pipeline_controlnet.py +134 -87
  144. diffusers/pipelines/controlnet/pipeline_controlnet_blip_diffusion.py +2 -2
  145. diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py +93 -77
  146. diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py +88 -197
  147. diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py +136 -90
  148. diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py +176 -80
  149. diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py +125 -89
  150. diffusers/pipelines/controlnet/pipeline_controlnet_union_inpaint_sd_xl.py +1790 -0
  151. diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl.py +1501 -0
  152. diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl_img2img.py +1627 -0
  153. diffusers/pipelines/controlnet/pipeline_flax_controlnet.py +2 -2
  154. diffusers/pipelines/controlnet_hunyuandit/__init__.py +48 -0
  155. diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py +1060 -0
  156. diffusers/pipelines/controlnet_sd3/__init__.py +57 -0
  157. diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet.py +1133 -0
  158. diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet_inpainting.py +1153 -0
  159. diffusers/pipelines/controlnet_xs/__init__.py +68 -0
  160. diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs.py +916 -0
  161. diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs_sd_xl.py +1111 -0
  162. diffusers/pipelines/ddpm/pipeline_ddpm.py +2 -2
  163. diffusers/pipelines/deepfloyd_if/pipeline_if.py +16 -30
  164. diffusers/pipelines/deepfloyd_if/pipeline_if_img2img.py +20 -35
  165. diffusers/pipelines/deepfloyd_if/pipeline_if_img2img_superresolution.py +23 -41
  166. diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting.py +22 -38
  167. diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting_superresolution.py +25 -41
  168. diffusers/pipelines/deepfloyd_if/pipeline_if_superresolution.py +19 -34
  169. diffusers/pipelines/deepfloyd_if/pipeline_output.py +6 -5
  170. diffusers/pipelines/deepfloyd_if/watermark.py +1 -1
  171. diffusers/pipelines/deprecated/alt_diffusion/modeling_roberta_series.py +11 -11
  172. diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py +70 -30
  173. diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py +48 -25
  174. diffusers/pipelines/deprecated/repaint/pipeline_repaint.py +2 -2
  175. diffusers/pipelines/deprecated/spectrogram_diffusion/pipeline_spectrogram_diffusion.py +7 -7
  176. diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_cycle_diffusion.py +21 -20
  177. diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_inpaint_legacy.py +27 -29
  178. diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_model_editing.py +33 -27
  179. diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_paradigms.py +33 -23
  180. diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_pix2pix_zero.py +36 -30
  181. diffusers/pipelines/deprecated/versatile_diffusion/modeling_text_unet.py +102 -69
  182. diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion.py +13 -13
  183. diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion_dual_guided.py +10 -5
  184. diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion_image_variation.py +11 -6
  185. diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion_text_to_image.py +10 -5
  186. diffusers/pipelines/deprecated/vq_diffusion/pipeline_vq_diffusion.py +5 -5
  187. diffusers/pipelines/dit/pipeline_dit.py +7 -4
  188. diffusers/pipelines/flux/__init__.py +69 -0
  189. diffusers/pipelines/flux/modeling_flux.py +47 -0
  190. diffusers/pipelines/flux/pipeline_flux.py +957 -0
  191. diffusers/pipelines/flux/pipeline_flux_control.py +889 -0
  192. diffusers/pipelines/flux/pipeline_flux_control_img2img.py +945 -0
  193. diffusers/pipelines/flux/pipeline_flux_control_inpaint.py +1141 -0
  194. diffusers/pipelines/flux/pipeline_flux_controlnet.py +1006 -0
  195. diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py +998 -0
  196. diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py +1204 -0
  197. diffusers/pipelines/flux/pipeline_flux_fill.py +969 -0
  198. diffusers/pipelines/flux/pipeline_flux_img2img.py +856 -0
  199. diffusers/pipelines/flux/pipeline_flux_inpaint.py +1022 -0
  200. diffusers/pipelines/flux/pipeline_flux_prior_redux.py +492 -0
  201. diffusers/pipelines/flux/pipeline_output.py +37 -0
  202. diffusers/pipelines/free_init_utils.py +41 -38
  203. diffusers/pipelines/free_noise_utils.py +596 -0
  204. diffusers/pipelines/hunyuan_video/__init__.py +48 -0
  205. diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video.py +687 -0
  206. diffusers/pipelines/hunyuan_video/pipeline_output.py +20 -0
  207. diffusers/pipelines/hunyuandit/__init__.py +48 -0
  208. diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py +916 -0
  209. diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py +33 -48
  210. diffusers/pipelines/kandinsky/pipeline_kandinsky.py +8 -8
  211. diffusers/pipelines/kandinsky/pipeline_kandinsky_combined.py +32 -29
  212. diffusers/pipelines/kandinsky/pipeline_kandinsky_img2img.py +11 -11
  213. diffusers/pipelines/kandinsky/pipeline_kandinsky_inpaint.py +12 -12
  214. diffusers/pipelines/kandinsky/pipeline_kandinsky_prior.py +10 -10
  215. diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2.py +6 -6
  216. diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_combined.py +34 -31
  217. diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_controlnet.py +10 -10
  218. diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_controlnet_img2img.py +10 -10
  219. diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_img2img.py +6 -6
  220. diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_inpainting.py +8 -8
  221. diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior.py +7 -7
  222. diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior_emb2emb.py +6 -6
  223. diffusers/pipelines/kandinsky3/convert_kandinsky3_unet.py +3 -3
  224. diffusers/pipelines/kandinsky3/pipeline_kandinsky3.py +22 -35
  225. diffusers/pipelines/kandinsky3/pipeline_kandinsky3_img2img.py +26 -37
  226. diffusers/pipelines/kolors/__init__.py +54 -0
  227. diffusers/pipelines/kolors/pipeline_kolors.py +1070 -0
  228. diffusers/pipelines/kolors/pipeline_kolors_img2img.py +1250 -0
  229. diffusers/pipelines/kolors/pipeline_output.py +21 -0
  230. diffusers/pipelines/kolors/text_encoder.py +889 -0
  231. diffusers/pipelines/kolors/tokenizer.py +338 -0
  232. diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_img2img.py +82 -62
  233. diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_text2img.py +77 -60
  234. diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py +12 -12
  235. diffusers/pipelines/latte/__init__.py +48 -0
  236. diffusers/pipelines/latte/pipeline_latte.py +881 -0
  237. diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py +80 -74
  238. diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py +85 -76
  239. diffusers/pipelines/ledits_pp/pipeline_output.py +2 -2
  240. diffusers/pipelines/ltx/__init__.py +50 -0
  241. diffusers/pipelines/ltx/pipeline_ltx.py +789 -0
  242. diffusers/pipelines/ltx/pipeline_ltx_image2video.py +885 -0
  243. diffusers/pipelines/ltx/pipeline_output.py +20 -0
  244. diffusers/pipelines/lumina/__init__.py +48 -0
  245. diffusers/pipelines/lumina/pipeline_lumina.py +890 -0
  246. diffusers/pipelines/marigold/__init__.py +50 -0
  247. diffusers/pipelines/marigold/marigold_image_processing.py +576 -0
  248. diffusers/pipelines/marigold/pipeline_marigold_depth.py +813 -0
  249. diffusers/pipelines/marigold/pipeline_marigold_normals.py +690 -0
  250. diffusers/pipelines/mochi/__init__.py +48 -0
  251. diffusers/pipelines/mochi/pipeline_mochi.py +748 -0
  252. diffusers/pipelines/mochi/pipeline_output.py +20 -0
  253. diffusers/pipelines/musicldm/pipeline_musicldm.py +14 -14
  254. diffusers/pipelines/pag/__init__.py +80 -0
  255. diffusers/pipelines/pag/pag_utils.py +243 -0
  256. diffusers/pipelines/pag/pipeline_pag_controlnet_sd.py +1328 -0
  257. diffusers/pipelines/pag/pipeline_pag_controlnet_sd_inpaint.py +1543 -0
  258. diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl.py +1610 -0
  259. diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl_img2img.py +1683 -0
  260. diffusers/pipelines/pag/pipeline_pag_hunyuandit.py +969 -0
  261. diffusers/pipelines/pag/pipeline_pag_kolors.py +1136 -0
  262. diffusers/pipelines/pag/pipeline_pag_pixart_sigma.py +865 -0
  263. diffusers/pipelines/pag/pipeline_pag_sana.py +886 -0
  264. diffusers/pipelines/pag/pipeline_pag_sd.py +1062 -0
  265. diffusers/pipelines/pag/pipeline_pag_sd_3.py +994 -0
  266. diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py +1058 -0
  267. diffusers/pipelines/pag/pipeline_pag_sd_animatediff.py +866 -0
  268. diffusers/pipelines/pag/pipeline_pag_sd_img2img.py +1094 -0
  269. diffusers/pipelines/pag/pipeline_pag_sd_inpaint.py +1356 -0
  270. diffusers/pipelines/pag/pipeline_pag_sd_xl.py +1345 -0
  271. diffusers/pipelines/pag/pipeline_pag_sd_xl_img2img.py +1544 -0
  272. diffusers/pipelines/pag/pipeline_pag_sd_xl_inpaint.py +1776 -0
  273. diffusers/pipelines/paint_by_example/pipeline_paint_by_example.py +17 -12
  274. diffusers/pipelines/pia/pipeline_pia.py +74 -164
  275. diffusers/pipelines/pipeline_flax_utils.py +5 -10
  276. diffusers/pipelines/pipeline_loading_utils.py +515 -53
  277. diffusers/pipelines/pipeline_utils.py +411 -222
  278. diffusers/pipelines/pixart_alpha/__init__.py +8 -1
  279. diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py +76 -93
  280. diffusers/pipelines/pixart_alpha/pipeline_pixart_sigma.py +873 -0
  281. diffusers/pipelines/sana/__init__.py +47 -0
  282. diffusers/pipelines/sana/pipeline_output.py +21 -0
  283. diffusers/pipelines/sana/pipeline_sana.py +884 -0
  284. diffusers/pipelines/semantic_stable_diffusion/pipeline_semantic_stable_diffusion.py +27 -23
  285. diffusers/pipelines/shap_e/pipeline_shap_e.py +3 -3
  286. diffusers/pipelines/shap_e/pipeline_shap_e_img2img.py +14 -14
  287. diffusers/pipelines/shap_e/renderer.py +1 -1
  288. diffusers/pipelines/stable_audio/__init__.py +50 -0
  289. diffusers/pipelines/stable_audio/modeling_stable_audio.py +158 -0
  290. diffusers/pipelines/stable_audio/pipeline_stable_audio.py +756 -0
  291. diffusers/pipelines/stable_cascade/pipeline_stable_cascade.py +71 -25
  292. diffusers/pipelines/stable_cascade/pipeline_stable_cascade_combined.py +23 -19
  293. diffusers/pipelines/stable_cascade/pipeline_stable_cascade_prior.py +35 -34
  294. diffusers/pipelines/stable_diffusion/__init__.py +0 -1
  295. diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +20 -11
  296. diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion.py +1 -1
  297. diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py +2 -2
  298. diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_upscale.py +6 -6
  299. diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py +145 -79
  300. diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py +43 -28
  301. diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_image_variation.py +13 -8
  302. diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py +100 -68
  303. diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py +109 -201
  304. diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py +131 -32
  305. diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_latent_upscale.py +247 -87
  306. diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py +30 -29
  307. diffusers/pipelines/stable_diffusion/pipeline_stable_unclip.py +35 -27
  308. diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py +49 -42
  309. diffusers/pipelines/stable_diffusion/safety_checker.py +2 -1
  310. diffusers/pipelines/stable_diffusion_3/__init__.py +54 -0
  311. diffusers/pipelines/stable_diffusion_3/pipeline_output.py +21 -0
  312. diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py +1140 -0
  313. diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py +1036 -0
  314. diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py +1250 -0
  315. diffusers/pipelines/stable_diffusion_attend_and_excite/pipeline_stable_diffusion_attend_and_excite.py +29 -20
  316. diffusers/pipelines/stable_diffusion_diffedit/pipeline_stable_diffusion_diffedit.py +59 -58
  317. diffusers/pipelines/stable_diffusion_gligen/pipeline_stable_diffusion_gligen.py +31 -25
  318. diffusers/pipelines/stable_diffusion_gligen/pipeline_stable_diffusion_gligen_text_image.py +38 -22
  319. diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_k_diffusion.py +30 -24
  320. diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py +24 -23
  321. diffusers/pipelines/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py +107 -67
  322. diffusers/pipelines/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py +316 -69
  323. diffusers/pipelines/stable_diffusion_safe/pipeline_stable_diffusion_safe.py +10 -5
  324. diffusers/pipelines/stable_diffusion_safe/safety_checker.py +1 -1
  325. diffusers/pipelines/stable_diffusion_sag/pipeline_stable_diffusion_sag.py +98 -30
  326. diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py +121 -83
  327. diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py +161 -105
  328. diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py +142 -218
  329. diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_instruct_pix2pix.py +45 -29
  330. diffusers/pipelines/stable_diffusion_xl/watermark.py +9 -3
  331. diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py +110 -57
  332. diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_adapter.py +69 -39
  333. diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_xl_adapter.py +105 -74
  334. diffusers/pipelines/text_to_video_synthesis/pipeline_output.py +3 -2
  335. diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth.py +29 -49
  336. diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth_img2img.py +32 -93
  337. diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero.py +37 -25
  338. diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero_sdxl.py +54 -40
  339. diffusers/pipelines/unclip/pipeline_unclip.py +6 -6
  340. diffusers/pipelines/unclip/pipeline_unclip_image_variation.py +6 -6
  341. diffusers/pipelines/unidiffuser/modeling_text_decoder.py +1 -1
  342. diffusers/pipelines/unidiffuser/modeling_uvit.py +12 -12
  343. diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py +29 -28
  344. diffusers/pipelines/wuerstchen/modeling_paella_vq_model.py +5 -5
  345. diffusers/pipelines/wuerstchen/modeling_wuerstchen_common.py +5 -10
  346. diffusers/pipelines/wuerstchen/modeling_wuerstchen_prior.py +6 -8
  347. diffusers/pipelines/wuerstchen/pipeline_wuerstchen.py +4 -4
  348. diffusers/pipelines/wuerstchen/pipeline_wuerstchen_combined.py +12 -12
  349. diffusers/pipelines/wuerstchen/pipeline_wuerstchen_prior.py +15 -14
  350. diffusers/{models/dual_transformer_2d.py → quantizers/__init__.py} +2 -6
  351. diffusers/quantizers/auto.py +139 -0
  352. diffusers/quantizers/base.py +233 -0
  353. diffusers/quantizers/bitsandbytes/__init__.py +2 -0
  354. diffusers/quantizers/bitsandbytes/bnb_quantizer.py +561 -0
  355. diffusers/quantizers/bitsandbytes/utils.py +306 -0
  356. diffusers/quantizers/gguf/__init__.py +1 -0
  357. diffusers/quantizers/gguf/gguf_quantizer.py +159 -0
  358. diffusers/quantizers/gguf/utils.py +456 -0
  359. diffusers/quantizers/quantization_config.py +669 -0
  360. diffusers/quantizers/torchao/__init__.py +15 -0
  361. diffusers/quantizers/torchao/torchao_quantizer.py +292 -0
  362. diffusers/schedulers/__init__.py +12 -2
  363. diffusers/schedulers/deprecated/__init__.py +1 -1
  364. diffusers/schedulers/deprecated/scheduling_karras_ve.py +25 -25
  365. diffusers/schedulers/scheduling_amused.py +5 -5
  366. diffusers/schedulers/scheduling_consistency_decoder.py +11 -11
  367. diffusers/schedulers/scheduling_consistency_models.py +23 -25
  368. diffusers/schedulers/scheduling_cosine_dpmsolver_multistep.py +572 -0
  369. diffusers/schedulers/scheduling_ddim.py +27 -26
  370. diffusers/schedulers/scheduling_ddim_cogvideox.py +452 -0
  371. diffusers/schedulers/scheduling_ddim_flax.py +2 -1
  372. diffusers/schedulers/scheduling_ddim_inverse.py +16 -16
  373. diffusers/schedulers/scheduling_ddim_parallel.py +32 -31
  374. diffusers/schedulers/scheduling_ddpm.py +27 -30
  375. diffusers/schedulers/scheduling_ddpm_flax.py +7 -3
  376. diffusers/schedulers/scheduling_ddpm_parallel.py +33 -36
  377. diffusers/schedulers/scheduling_ddpm_wuerstchen.py +14 -14
  378. diffusers/schedulers/scheduling_deis_multistep.py +150 -50
  379. diffusers/schedulers/scheduling_dpm_cogvideox.py +489 -0
  380. diffusers/schedulers/scheduling_dpmsolver_multistep.py +221 -84
  381. diffusers/schedulers/scheduling_dpmsolver_multistep_flax.py +2 -2
  382. diffusers/schedulers/scheduling_dpmsolver_multistep_inverse.py +158 -52
  383. diffusers/schedulers/scheduling_dpmsolver_sde.py +153 -34
  384. diffusers/schedulers/scheduling_dpmsolver_singlestep.py +275 -86
  385. diffusers/schedulers/scheduling_edm_dpmsolver_multistep.py +81 -57
  386. diffusers/schedulers/scheduling_edm_euler.py +62 -39
  387. diffusers/schedulers/scheduling_euler_ancestral_discrete.py +30 -29
  388. diffusers/schedulers/scheduling_euler_discrete.py +255 -74
  389. diffusers/schedulers/scheduling_flow_match_euler_discrete.py +458 -0
  390. diffusers/schedulers/scheduling_flow_match_heun_discrete.py +320 -0
  391. diffusers/schedulers/scheduling_heun_discrete.py +174 -46
  392. diffusers/schedulers/scheduling_ipndm.py +9 -9
  393. diffusers/schedulers/scheduling_k_dpm_2_ancestral_discrete.py +138 -29
  394. diffusers/schedulers/scheduling_k_dpm_2_discrete.py +132 -26
  395. diffusers/schedulers/scheduling_karras_ve_flax.py +6 -6
  396. diffusers/schedulers/scheduling_lcm.py +23 -29
  397. diffusers/schedulers/scheduling_lms_discrete.py +105 -28
  398. diffusers/schedulers/scheduling_pndm.py +20 -20
  399. diffusers/schedulers/scheduling_repaint.py +21 -21
  400. diffusers/schedulers/scheduling_sasolver.py +157 -60
  401. diffusers/schedulers/scheduling_sde_ve.py +19 -19
  402. diffusers/schedulers/scheduling_tcd.py +41 -36
  403. diffusers/schedulers/scheduling_unclip.py +19 -16
  404. diffusers/schedulers/scheduling_unipc_multistep.py +243 -47
  405. diffusers/schedulers/scheduling_utils.py +12 -5
  406. diffusers/schedulers/scheduling_utils_flax.py +1 -3
  407. diffusers/schedulers/scheduling_vq_diffusion.py +10 -10
  408. diffusers/training_utils.py +214 -30
  409. diffusers/utils/__init__.py +17 -1
  410. diffusers/utils/constants.py +3 -0
  411. diffusers/utils/doc_utils.py +1 -0
  412. diffusers/utils/dummy_pt_objects.py +592 -7
  413. diffusers/utils/dummy_torch_and_torchsde_objects.py +15 -0
  414. diffusers/utils/dummy_torch_and_transformers_and_sentencepiece_objects.py +47 -0
  415. diffusers/utils/dummy_torch_and_transformers_objects.py +1001 -71
  416. diffusers/utils/dynamic_modules_utils.py +34 -29
  417. diffusers/utils/export_utils.py +50 -6
  418. diffusers/utils/hub_utils.py +131 -17
  419. diffusers/utils/import_utils.py +210 -8
  420. diffusers/utils/loading_utils.py +118 -5
  421. diffusers/utils/logging.py +4 -2
  422. diffusers/utils/peft_utils.py +37 -7
  423. diffusers/utils/state_dict_utils.py +13 -2
  424. diffusers/utils/testing_utils.py +193 -11
  425. diffusers/utils/torch_utils.py +4 -0
  426. diffusers/video_processor.py +113 -0
  427. {diffusers-0.27.0.dist-info → diffusers-0.32.2.dist-info}/METADATA +82 -91
  428. diffusers-0.32.2.dist-info/RECORD +550 -0
  429. {diffusers-0.27.0.dist-info → diffusers-0.32.2.dist-info}/WHEEL +1 -1
  430. diffusers/loaders/autoencoder.py +0 -146
  431. diffusers/loaders/controlnet.py +0 -136
  432. diffusers/loaders/lora.py +0 -1349
  433. diffusers/models/prior_transformer.py +0 -12
  434. diffusers/models/t5_film_transformer.py +0 -70
  435. diffusers/models/transformer_2d.py +0 -25
  436. diffusers/models/transformer_temporal.py +0 -34
  437. diffusers/models/unet_1d.py +0 -26
  438. diffusers/models/unet_1d_blocks.py +0 -203
  439. diffusers/models/unet_2d.py +0 -27
  440. diffusers/models/unet_2d_blocks.py +0 -375
  441. diffusers/models/unet_2d_condition.py +0 -25
  442. diffusers-0.27.0.dist-info/RECORD +0 -399
  443. {diffusers-0.27.0.dist-info → diffusers-0.32.2.dist-info}/LICENSE +0 -0
  444. {diffusers-0.27.0.dist-info → diffusers-0.32.2.dist-info}/entry_points.txt +0 -0
  445. {diffusers-0.27.0.dist-info → diffusers-0.32.2.dist-info}/top_level.txt +0 -0
diffusers/loaders/peft.py CHANGED
@@ -12,15 +12,104 @@
12
12
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
13
  # See the License for the specific language governing permissions and
14
14
  # limitations under the License.
15
- from typing import List, Union
16
-
17
- from ..utils import MIN_PEFT_VERSION, check_peft_version, is_peft_available
15
+ import inspect
16
+ import os
17
+ from functools import partial
18
+ from pathlib import Path
19
+ from typing import Dict, List, Optional, Union
20
+
21
+ import safetensors
22
+ import torch
23
+
24
+ from ..utils import (
25
+ MIN_PEFT_VERSION,
26
+ USE_PEFT_BACKEND,
27
+ check_peft_version,
28
+ convert_unet_state_dict_to_peft,
29
+ delete_adapter_layers,
30
+ get_adapter_name,
31
+ get_peft_kwargs,
32
+ is_peft_available,
33
+ is_peft_version,
34
+ logging,
35
+ set_adapter_layers,
36
+ set_weights_and_activate_adapters,
37
+ )
38
+ from .lora_base import _fetch_state_dict, _func_optionally_disable_offloading
39
+ from .unet_loader_utils import _maybe_expand_lora_scales
40
+
41
+
42
+ logger = logging.get_logger(__name__)
43
+
44
+ _SET_ADAPTER_SCALE_FN_MAPPING = {
45
+ "UNet2DConditionModel": _maybe_expand_lora_scales,
46
+ "UNetMotionModel": _maybe_expand_lora_scales,
47
+ "SD3Transformer2DModel": lambda model_cls, weights: weights,
48
+ "FluxTransformer2DModel": lambda model_cls, weights: weights,
49
+ "CogVideoXTransformer3DModel": lambda model_cls, weights: weights,
50
+ "MochiTransformer3DModel": lambda model_cls, weights: weights,
51
+ "HunyuanVideoTransformer3DModel": lambda model_cls, weights: weights,
52
+ "LTXVideoTransformer3DModel": lambda model_cls, weights: weights,
53
+ "SanaTransformer2DModel": lambda model_cls, weights: weights,
54
+ }
55
+
56
+
57
+ def _maybe_adjust_config(config):
58
+ """
59
+ We may run into some ambiguous configuration values when a model has module names, sharing a common prefix
60
+ (`proj_out.weight` and `blocks.transformer.proj_out.weight`, for example) and they have different LoRA ranks. This
61
+ method removes the ambiguity by following what is described here:
62
+ https://github.com/huggingface/diffusers/pull/9985#issuecomment-2493840028.
63
+ """
64
+ rank_pattern = config["rank_pattern"].copy()
65
+ target_modules = config["target_modules"]
66
+ original_r = config["r"]
67
+
68
+ for key in list(rank_pattern.keys()):
69
+ key_rank = rank_pattern[key]
70
+
71
+ # try to detect ambiguity
72
+ # `target_modules` can also be a str, in which case this loop would loop
73
+ # over the chars of the str. The technically correct way to match LoRA keys
74
+ # in PEFT is to use LoraModel._check_target_module_exists (lora_config, key).
75
+ # But this cuts it for now.
76
+ exact_matches = [mod for mod in target_modules if mod == key]
77
+ substring_matches = [mod for mod in target_modules if key in mod and mod != key]
78
+ ambiguous_key = key
79
+
80
+ if exact_matches and substring_matches:
81
+ # if ambiguous we update the rank associated with the ambiguous key (`proj_out`, for example)
82
+ config["r"] = key_rank
83
+ # remove the ambiguous key from `rank_pattern` and update its rank to `r`, instead
84
+ del config["rank_pattern"][key]
85
+ for mod in substring_matches:
86
+ # avoid overwriting if the module already has a specific rank
87
+ if mod not in config["rank_pattern"]:
88
+ config["rank_pattern"][mod] = original_r
89
+
90
+ # update the rest of the keys with the `original_r`
91
+ for mod in target_modules:
92
+ if mod != ambiguous_key and mod not in config["rank_pattern"]:
93
+ config["rank_pattern"][mod] = original_r
94
+
95
+ # handle alphas to deal with cases like
96
+ # https://github.com/huggingface/diffusers/pull/9999#issuecomment-2516180777
97
+ has_different_ranks = len(config["rank_pattern"]) > 1 and list(config["rank_pattern"])[0] != config["r"]
98
+ if has_different_ranks:
99
+ config["lora_alpha"] = config["r"]
100
+ alpha_pattern = {}
101
+ for module_name, rank in config["rank_pattern"].items():
102
+ alpha_pattern[module_name] = rank
103
+ config["alpha_pattern"] = alpha_pattern
104
+
105
+ return config
18
106
 
19
107
 
20
108
  class PeftAdapterMixin:
21
109
  """
22
110
  A class containing all functions for loading and using adapters weights that are supported in PEFT library. For
23
- more details about adapters and injecting them in a transformer-based model, check out the PEFT [documentation](https://huggingface.co/docs/peft/index).
111
+ more details about adapters and injecting them in a base model, check out the PEFT
112
+ [documentation](https://huggingface.co/docs/peft/index).
24
113
 
25
114
  Install the latest version of PEFT, and use this mixin to:
26
115
 
@@ -32,6 +121,348 @@ class PeftAdapterMixin:
32
121
 
33
122
  _hf_peft_config_loaded = False
34
123
 
124
+ @classmethod
125
+ # Copied from diffusers.loaders.lora_base.LoraBaseMixin._optionally_disable_offloading
126
+ def _optionally_disable_offloading(cls, _pipeline):
127
+ """
128
+ Optionally removes offloading in case the pipeline has been already sequentially offloaded to CPU.
129
+
130
+ Args:
131
+ _pipeline (`DiffusionPipeline`):
132
+ The pipeline to disable offloading for.
133
+
134
+ Returns:
135
+ tuple:
136
+ A tuple indicating if `is_model_cpu_offload` or `is_sequential_cpu_offload` is True.
137
+ """
138
+ return _func_optionally_disable_offloading(_pipeline=_pipeline)
139
+
140
+ def load_lora_adapter(self, pretrained_model_name_or_path_or_dict, prefix="transformer", **kwargs):
141
+ r"""
142
+ Loads a LoRA adapter into the underlying model.
143
+
144
+ Parameters:
145
+ pretrained_model_name_or_path_or_dict (`str` or `os.PathLike` or `dict`):
146
+ Can be either:
147
+
148
+ - A string, the *model id* (for example `google/ddpm-celebahq-256`) of a pretrained model hosted on
149
+ the Hub.
150
+ - A path to a *directory* (for example `./my_model_directory`) containing the model weights saved
151
+ with [`ModelMixin.save_pretrained`].
152
+ - A [torch state
153
+ dict](https://pytorch.org/tutorials/beginner/saving_loading_models.html#what-is-a-state-dict).
154
+
155
+ prefix (`str`, *optional*): Prefix to filter the state dict.
156
+
157
+ cache_dir (`Union[str, os.PathLike]`, *optional*):
158
+ Path to a directory where a downloaded pretrained model configuration is cached if the standard cache
159
+ is not used.
160
+ force_download (`bool`, *optional*, defaults to `False`):
161
+ Whether or not to force the (re-)download of the model weights and configuration files, overriding the
162
+ cached versions if they exist.
163
+ proxies (`Dict[str, str]`, *optional*):
164
+ A dictionary of proxy servers to use by protocol or endpoint, for example, `{'http': 'foo.bar:3128',
165
+ 'http://hostname': 'foo.bar:4012'}`. The proxies are used on each request.
166
+ local_files_only (`bool`, *optional*, defaults to `False`):
167
+ Whether to only load local model weights and configuration files or not. If set to `True`, the model
168
+ won't be downloaded from the Hub.
169
+ token (`str` or *bool*, *optional*):
170
+ The token to use as HTTP bearer authorization for remote files. If `True`, the token generated from
171
+ `diffusers-cli login` (stored in `~/.huggingface`) is used.
172
+ revision (`str`, *optional*, defaults to `"main"`):
173
+ The specific model version to use. It can be a branch name, a tag name, a commit id, or any identifier
174
+ allowed by Git.
175
+ subfolder (`str`, *optional*, defaults to `""`):
176
+ The subfolder location of a model file within a larger model repository on the Hub or locally.
177
+ network_alphas (`Dict[str, float]`):
178
+ The value of the network alpha used for stable learning and preventing underflow. This value has the
179
+ same meaning as the `--network_alpha` option in the kohya-ss trainer script. Refer to [this
180
+ link](https://github.com/darkstorm2150/sd-scripts/blob/main/docs/train_network_README-en.md#execute-learning).
181
+ low_cpu_mem_usage (`bool`, *optional*):
182
+ Speed up model loading by only loading the pretrained LoRA weights and not initializing the random
183
+ weights.
184
+ """
185
+ from peft import LoraConfig, inject_adapter_in_model, set_peft_model_state_dict
186
+ from peft.tuners.tuners_utils import BaseTunerLayer
187
+
188
+ cache_dir = kwargs.pop("cache_dir", None)
189
+ force_download = kwargs.pop("force_download", False)
190
+ proxies = kwargs.pop("proxies", None)
191
+ local_files_only = kwargs.pop("local_files_only", None)
192
+ token = kwargs.pop("token", None)
193
+ revision = kwargs.pop("revision", None)
194
+ subfolder = kwargs.pop("subfolder", None)
195
+ weight_name = kwargs.pop("weight_name", None)
196
+ use_safetensors = kwargs.pop("use_safetensors", None)
197
+ adapter_name = kwargs.pop("adapter_name", None)
198
+ network_alphas = kwargs.pop("network_alphas", None)
199
+ _pipeline = kwargs.pop("_pipeline", None)
200
+ low_cpu_mem_usage = kwargs.pop("low_cpu_mem_usage", False)
201
+ allow_pickle = False
202
+
203
+ if low_cpu_mem_usage and is_peft_version("<=", "0.13.0"):
204
+ raise ValueError(
205
+ "`low_cpu_mem_usage=True` is not compatible with this `peft` version. Please update it with `pip install -U peft`."
206
+ )
207
+
208
+ user_agent = {
209
+ "file_type": "attn_procs_weights",
210
+ "framework": "pytorch",
211
+ }
212
+
213
+ state_dict = _fetch_state_dict(
214
+ pretrained_model_name_or_path_or_dict=pretrained_model_name_or_path_or_dict,
215
+ weight_name=weight_name,
216
+ use_safetensors=use_safetensors,
217
+ local_files_only=local_files_only,
218
+ cache_dir=cache_dir,
219
+ force_download=force_download,
220
+ proxies=proxies,
221
+ token=token,
222
+ revision=revision,
223
+ subfolder=subfolder,
224
+ user_agent=user_agent,
225
+ allow_pickle=allow_pickle,
226
+ )
227
+ if network_alphas is not None and prefix is None:
228
+ raise ValueError("`network_alphas` cannot be None when `prefix` is None.")
229
+
230
+ if prefix is not None:
231
+ keys = list(state_dict.keys())
232
+ model_keys = [k for k in keys if k.startswith(f"{prefix}.")]
233
+ if len(model_keys) > 0:
234
+ state_dict = {k.replace(f"{prefix}.", ""): v for k, v in state_dict.items() if k in model_keys}
235
+
236
+ if len(state_dict) > 0:
237
+ if adapter_name in getattr(self, "peft_config", {}):
238
+ raise ValueError(
239
+ f"Adapter name {adapter_name} already in use in the model - please select a new adapter name."
240
+ )
241
+
242
+ # check with first key if is not in peft format
243
+ first_key = next(iter(state_dict.keys()))
244
+ if "lora_A" not in first_key:
245
+ state_dict = convert_unet_state_dict_to_peft(state_dict)
246
+
247
+ rank = {}
248
+ for key, val in state_dict.items():
249
+ # Cannot figure out rank from lora layers that don't have atleast 2 dimensions.
250
+ # Bias layers in LoRA only have a single dimension
251
+ if "lora_B" in key and val.ndim > 1:
252
+ rank[key] = val.shape[1]
253
+
254
+ if network_alphas is not None and len(network_alphas) >= 1:
255
+ alpha_keys = [k for k in network_alphas.keys() if k.startswith(f"{prefix}.")]
256
+ network_alphas = {k.replace(f"{prefix}.", ""): v for k, v in network_alphas.items() if k in alpha_keys}
257
+
258
+ lora_config_kwargs = get_peft_kwargs(rank, network_alpha_dict=network_alphas, peft_state_dict=state_dict)
259
+ lora_config_kwargs = _maybe_adjust_config(lora_config_kwargs)
260
+
261
+ if "use_dora" in lora_config_kwargs:
262
+ if lora_config_kwargs["use_dora"]:
263
+ if is_peft_version("<", "0.9.0"):
264
+ raise ValueError(
265
+ "You need `peft` 0.9.0 at least to use DoRA-enabled LoRAs. Please upgrade your installation of `peft`."
266
+ )
267
+ else:
268
+ if is_peft_version("<", "0.9.0"):
269
+ lora_config_kwargs.pop("use_dora")
270
+
271
+ if "lora_bias" in lora_config_kwargs:
272
+ if lora_config_kwargs["lora_bias"]:
273
+ if is_peft_version("<=", "0.13.2"):
274
+ raise ValueError(
275
+ "You need `peft` 0.14.0 at least to use `lora_bias` in LoRAs. Please upgrade your installation of `peft`."
276
+ )
277
+ else:
278
+ if is_peft_version("<=", "0.13.2"):
279
+ lora_config_kwargs.pop("lora_bias")
280
+
281
+ lora_config = LoraConfig(**lora_config_kwargs)
282
+ # adapter_name
283
+ if adapter_name is None:
284
+ adapter_name = get_adapter_name(self)
285
+
286
+ # <Unsafe code
287
+ # We can be sure that the following works as it just sets attention processors, lora layers and puts all in the same dtype
288
+ # Now we remove any existing hooks to `_pipeline`.
289
+
290
+ # In case the pipeline has been already offloaded to CPU - temporarily remove the hooks
291
+ # otherwise loading LoRA weights will lead to an error
292
+ is_model_cpu_offload, is_sequential_cpu_offload = self._optionally_disable_offloading(_pipeline)
293
+
294
+ peft_kwargs = {}
295
+ if is_peft_version(">=", "0.13.1"):
296
+ peft_kwargs["low_cpu_mem_usage"] = low_cpu_mem_usage
297
+
298
+ # To handle scenarios where we cannot successfully set state dict. If it's unsucessful,
299
+ # we should also delete the `peft_config` associated to the `adapter_name`.
300
+ try:
301
+ inject_adapter_in_model(lora_config, self, adapter_name=adapter_name, **peft_kwargs)
302
+ incompatible_keys = set_peft_model_state_dict(self, state_dict, adapter_name, **peft_kwargs)
303
+ except RuntimeError as e:
304
+ for module in self.modules():
305
+ if isinstance(module, BaseTunerLayer):
306
+ active_adapters = module.active_adapters
307
+ for active_adapter in active_adapters:
308
+ if adapter_name in active_adapter:
309
+ module.delete_adapter(adapter_name)
310
+
311
+ self.peft_config.pop(adapter_name)
312
+ logger.error(f"Loading {adapter_name} was unsucessful with the following error: \n{e}")
313
+ raise
314
+
315
+ warn_msg = ""
316
+ if incompatible_keys is not None:
317
+ # Check only for unexpected keys.
318
+ unexpected_keys = getattr(incompatible_keys, "unexpected_keys", None)
319
+ if unexpected_keys:
320
+ lora_unexpected_keys = [k for k in unexpected_keys if "lora_" in k and adapter_name in k]
321
+ if lora_unexpected_keys:
322
+ warn_msg = (
323
+ f"Loading adapter weights from state_dict led to unexpected keys found in the model:"
324
+ f" {', '.join(lora_unexpected_keys)}. "
325
+ )
326
+
327
+ # Filter missing keys specific to the current adapter.
328
+ missing_keys = getattr(incompatible_keys, "missing_keys", None)
329
+ if missing_keys:
330
+ lora_missing_keys = [k for k in missing_keys if "lora_" in k and adapter_name in k]
331
+ if lora_missing_keys:
332
+ warn_msg += (
333
+ f"Loading adapter weights from state_dict led to missing keys in the model:"
334
+ f" {', '.join(lora_missing_keys)}."
335
+ )
336
+
337
+ if warn_msg:
338
+ logger.warning(warn_msg)
339
+
340
+ # Offload back.
341
+ if is_model_cpu_offload:
342
+ _pipeline.enable_model_cpu_offload()
343
+ elif is_sequential_cpu_offload:
344
+ _pipeline.enable_sequential_cpu_offload()
345
+ # Unsafe code />
346
+
347
+ def save_lora_adapter(
348
+ self,
349
+ save_directory,
350
+ adapter_name: str = "default",
351
+ upcast_before_saving: bool = False,
352
+ safe_serialization: bool = True,
353
+ weight_name: Optional[str] = None,
354
+ ):
355
+ """
356
+ Save the LoRA parameters corresponding to the underlying model.
357
+
358
+ Arguments:
359
+ save_directory (`str` or `os.PathLike`):
360
+ Directory to save LoRA parameters to. Will be created if it doesn't exist.
361
+ adapter_name: (`str`, defaults to "default"): The name of the adapter to serialize. Useful when the
362
+ underlying model has multiple adapters loaded.
363
+ upcast_before_saving (`bool`, defaults to `False`):
364
+ Whether to cast the underlying model to `torch.float32` before serialization.
365
+ save_function (`Callable`):
366
+ The function to use to save the state dictionary. Useful during distributed training when you need to
367
+ replace `torch.save` with another method. Can be configured with the environment variable
368
+ `DIFFUSERS_SAVE_MODE`.
369
+ safe_serialization (`bool`, *optional*, defaults to `True`):
370
+ Whether to save the model using `safetensors` or the traditional PyTorch way with `pickle`.
371
+ weight_name: (`str`, *optional*, defaults to `None`): Name of the file to serialize the state dict with.
372
+ """
373
+ from peft.utils import get_peft_model_state_dict
374
+
375
+ from .lora_base import LORA_WEIGHT_NAME, LORA_WEIGHT_NAME_SAFE
376
+
377
+ if adapter_name is None:
378
+ adapter_name = get_adapter_name(self)
379
+
380
+ if adapter_name not in getattr(self, "peft_config", {}):
381
+ raise ValueError(f"Adapter name {adapter_name} not found in the model.")
382
+
383
+ lora_layers_to_save = get_peft_model_state_dict(
384
+ self.to(dtype=torch.float32 if upcast_before_saving else None), adapter_name=adapter_name
385
+ )
386
+ if os.path.isfile(save_directory):
387
+ raise ValueError(f"Provided path ({save_directory}) should be a directory, not a file")
388
+
389
+ if safe_serialization:
390
+
391
+ def save_function(weights, filename):
392
+ return safetensors.torch.save_file(weights, filename, metadata={"format": "pt"})
393
+
394
+ else:
395
+ save_function = torch.save
396
+
397
+ os.makedirs(save_directory, exist_ok=True)
398
+
399
+ if weight_name is None:
400
+ if safe_serialization:
401
+ weight_name = LORA_WEIGHT_NAME_SAFE
402
+ else:
403
+ weight_name = LORA_WEIGHT_NAME
404
+
405
+ # TODO: we could consider saving the `peft_config` as well.
406
+ save_path = Path(save_directory, weight_name).as_posix()
407
+ save_function(lora_layers_to_save, save_path)
408
+ logger.info(f"Model weights saved in {save_path}")
409
+
410
+ def set_adapters(
411
+ self,
412
+ adapter_names: Union[List[str], str],
413
+ weights: Optional[Union[float, Dict, List[float], List[Dict], List[None]]] = None,
414
+ ):
415
+ """
416
+ Set the currently active adapters for use in the UNet.
417
+
418
+ Args:
419
+ adapter_names (`List[str]` or `str`):
420
+ The names of the adapters to use.
421
+ adapter_weights (`Union[List[float], float]`, *optional*):
422
+ The adapter(s) weights to use with the UNet. If `None`, the weights are set to `1.0` for all the
423
+ adapters.
424
+
425
+ Example:
426
+
427
+ ```py
428
+ from diffusers import AutoPipelineForText2Image
429
+ import torch
430
+
431
+ pipeline = AutoPipelineForText2Image.from_pretrained(
432
+ "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16
433
+ ).to("cuda")
434
+ pipeline.load_lora_weights(
435
+ "jbilcke-hf/sdxl-cinematic-1", weight_name="pytorch_lora_weights.safetensors", adapter_name="cinematic"
436
+ )
437
+ pipeline.load_lora_weights("nerijs/pixel-art-xl", weight_name="pixel-art-xl.safetensors", adapter_name="pixel")
438
+ pipeline.set_adapters(["cinematic", "pixel"], adapter_weights=[0.5, 0.5])
439
+ ```
440
+ """
441
+ if not USE_PEFT_BACKEND:
442
+ raise ValueError("PEFT backend is required for `set_adapters()`.")
443
+
444
+ adapter_names = [adapter_names] if isinstance(adapter_names, str) else adapter_names
445
+
446
+ # Expand weights into a list, one entry per adapter
447
+ # examples for e.g. 2 adapters: [{...}, 7] -> [7,7] ; None -> [None, None]
448
+ if not isinstance(weights, list):
449
+ weights = [weights] * len(adapter_names)
450
+
451
+ if len(adapter_names) != len(weights):
452
+ raise ValueError(
453
+ f"Length of adapter names {len(adapter_names)} is not equal to the length of their weights {len(weights)}."
454
+ )
455
+
456
+ # Set None values to default of 1.0
457
+ # e.g. [{...}, 7] -> [{...}, 7] ; [None, None] -> [1.0, 1.0]
458
+ weights = [w if w is not None else 1.0 for w in weights]
459
+
460
+ # e.g. [{...}, 7] -> [{expanded dict...}, 7]
461
+ scale_expansion_fn = _SET_ADAPTER_SCALE_FN_MAPPING[self.__class__.__name__]
462
+ weights = scale_expansion_fn(self, weights)
463
+
464
+ set_weights_and_activate_adapters(self, adapter_names, weights)
465
+
35
466
  def add_adapter(self, adapter_config, adapter_name: str = "default") -> None:
36
467
  r"""
37
468
  Adds a new adapter to the current model for training. If no adapter name is passed, a default name is assigned
@@ -65,7 +496,7 @@ class PeftAdapterMixin:
65
496
  )
66
497
 
67
498
  # Unlike transformers, here we don't need to retrieve the name_or_path of the unet as the loading logic is
68
- # handled by the `load_lora_layers` or `LoraLoaderMixin`. Therefore we set it to `None` here.
499
+ # handled by the `load_lora_layers` or `StableDiffusionLoraLoaderMixin`. Therefore we set it to `None` here.
69
500
  adapter_config.base_model_name_or_path = None
70
501
  inject_adapter_in_model(adapter_config, self, adapter_name)
71
502
  self.set_adapter(adapter_name)
@@ -143,8 +574,8 @@ class PeftAdapterMixin:
143
574
 
144
575
  def enable_adapters(self) -> None:
145
576
  """
146
- Enable adapters that are attached to the model. The model uses `self.active_adapters()` to retrieve the
147
- list of adapters to enable.
577
+ Enable adapters that are attached to the model. The model uses `self.active_adapters()` to retrieve the list of
578
+ adapters to enable.
148
579
 
149
580
  If you are not familiar with adapters and PEFT methods, we invite you to read more about them on the PEFT
150
581
  [documentation](https://huggingface.co/docs/peft).
@@ -184,3 +615,136 @@ class PeftAdapterMixin:
184
615
  for _, module in self.named_modules():
185
616
  if isinstance(module, BaseTunerLayer):
186
617
  return module.active_adapter
618
+
619
+ def fuse_lora(self, lora_scale=1.0, safe_fusing=False, adapter_names=None):
620
+ if not USE_PEFT_BACKEND:
621
+ raise ValueError("PEFT backend is required for `fuse_lora()`.")
622
+
623
+ self.lora_scale = lora_scale
624
+ self._safe_fusing = safe_fusing
625
+ self.apply(partial(self._fuse_lora_apply, adapter_names=adapter_names))
626
+
627
+ def _fuse_lora_apply(self, module, adapter_names=None):
628
+ from peft.tuners.tuners_utils import BaseTunerLayer
629
+
630
+ merge_kwargs = {"safe_merge": self._safe_fusing}
631
+
632
+ if isinstance(module, BaseTunerLayer):
633
+ if self.lora_scale != 1.0:
634
+ module.scale_layer(self.lora_scale)
635
+
636
+ # For BC with prevous PEFT versions, we need to check the signature
637
+ # of the `merge` method to see if it supports the `adapter_names` argument.
638
+ supported_merge_kwargs = list(inspect.signature(module.merge).parameters)
639
+ if "adapter_names" in supported_merge_kwargs:
640
+ merge_kwargs["adapter_names"] = adapter_names
641
+ elif "adapter_names" not in supported_merge_kwargs and adapter_names is not None:
642
+ raise ValueError(
643
+ "The `adapter_names` argument is not supported with your PEFT version. Please upgrade"
644
+ " to the latest version of PEFT. `pip install -U peft`"
645
+ )
646
+
647
+ module.merge(**merge_kwargs)
648
+
649
+ def unfuse_lora(self):
650
+ if not USE_PEFT_BACKEND:
651
+ raise ValueError("PEFT backend is required for `unfuse_lora()`.")
652
+ self.apply(self._unfuse_lora_apply)
653
+
654
+ def _unfuse_lora_apply(self, module):
655
+ from peft.tuners.tuners_utils import BaseTunerLayer
656
+
657
+ if isinstance(module, BaseTunerLayer):
658
+ module.unmerge()
659
+
660
+ def unload_lora(self):
661
+ if not USE_PEFT_BACKEND:
662
+ raise ValueError("PEFT backend is required for `unload_lora()`.")
663
+
664
+ from ..utils import recurse_remove_peft_layers
665
+
666
+ recurse_remove_peft_layers(self)
667
+ if hasattr(self, "peft_config"):
668
+ del self.peft_config
669
+
670
+ def disable_lora(self):
671
+ """
672
+ Disables the active LoRA layers of the underlying model.
673
+
674
+ Example:
675
+
676
+ ```py
677
+ from diffusers import AutoPipelineForText2Image
678
+ import torch
679
+
680
+ pipeline = AutoPipelineForText2Image.from_pretrained(
681
+ "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16
682
+ ).to("cuda")
683
+ pipeline.load_lora_weights(
684
+ "jbilcke-hf/sdxl-cinematic-1", weight_name="pytorch_lora_weights.safetensors", adapter_name="cinematic"
685
+ )
686
+ pipeline.disable_lora()
687
+ ```
688
+ """
689
+ if not USE_PEFT_BACKEND:
690
+ raise ValueError("PEFT backend is required for this method.")
691
+ set_adapter_layers(self, enabled=False)
692
+
693
+ def enable_lora(self):
694
+ """
695
+ Enables the active LoRA layers of the underlying model.
696
+
697
+ Example:
698
+
699
+ ```py
700
+ from diffusers import AutoPipelineForText2Image
701
+ import torch
702
+
703
+ pipeline = AutoPipelineForText2Image.from_pretrained(
704
+ "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16
705
+ ).to("cuda")
706
+ pipeline.load_lora_weights(
707
+ "jbilcke-hf/sdxl-cinematic-1", weight_name="pytorch_lora_weights.safetensors", adapter_name="cinematic"
708
+ )
709
+ pipeline.enable_lora()
710
+ ```
711
+ """
712
+ if not USE_PEFT_BACKEND:
713
+ raise ValueError("PEFT backend is required for this method.")
714
+ set_adapter_layers(self, enabled=True)
715
+
716
+ def delete_adapters(self, adapter_names: Union[List[str], str]):
717
+ """
718
+ Delete an adapter's LoRA layers from the underlying model.
719
+
720
+ Args:
721
+ adapter_names (`Union[List[str], str]`):
722
+ The names (single string or list of strings) of the adapter to delete.
723
+
724
+ Example:
725
+
726
+ ```py
727
+ from diffusers import AutoPipelineForText2Image
728
+ import torch
729
+
730
+ pipeline = AutoPipelineForText2Image.from_pretrained(
731
+ "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16
732
+ ).to("cuda")
733
+ pipeline.load_lora_weights(
734
+ "jbilcke-hf/sdxl-cinematic-1", weight_name="pytorch_lora_weights.safetensors", adapter_names="cinematic"
735
+ )
736
+ pipeline.delete_adapters("cinematic")
737
+ ```
738
+ """
739
+ if not USE_PEFT_BACKEND:
740
+ raise ValueError("PEFT backend is required for this method.")
741
+
742
+ if isinstance(adapter_names, str):
743
+ adapter_names = [adapter_names]
744
+
745
+ for adapter_name in adapter_names:
746
+ delete_adapter_layers(self, adapter_name)
747
+
748
+ # Pop also the corresponding adapter from the config
749
+ if hasattr(self, "peft_config"):
750
+ self.peft_config.pop(adapter_name, None)