diffusers 0.27.1__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.
- diffusers/__init__.py +233 -6
- diffusers/callbacks.py +209 -0
- diffusers/commands/env.py +102 -6
- diffusers/configuration_utils.py +45 -16
- diffusers/dependency_versions_table.py +4 -3
- diffusers/image_processor.py +434 -110
- diffusers/loaders/__init__.py +42 -9
- diffusers/loaders/ip_adapter.py +626 -36
- diffusers/loaders/lora_base.py +900 -0
- diffusers/loaders/lora_conversion_utils.py +991 -125
- diffusers/loaders/lora_pipeline.py +3812 -0
- diffusers/loaders/peft.py +571 -7
- diffusers/loaders/single_file.py +405 -173
- diffusers/loaders/single_file_model.py +385 -0
- diffusers/loaders/single_file_utils.py +1783 -713
- diffusers/loaders/textual_inversion.py +41 -23
- diffusers/loaders/transformer_flux.py +181 -0
- diffusers/loaders/transformer_sd3.py +89 -0
- diffusers/loaders/unet.py +464 -540
- diffusers/loaders/unet_loader_utils.py +163 -0
- diffusers/models/__init__.py +76 -7
- diffusers/models/activations.py +65 -10
- diffusers/models/adapter.py +53 -53
- diffusers/models/attention.py +605 -18
- diffusers/models/attention_flax.py +1 -1
- diffusers/models/attention_processor.py +4304 -687
- diffusers/models/autoencoders/__init__.py +8 -0
- diffusers/models/autoencoders/autoencoder_asym_kl.py +15 -17
- diffusers/models/autoencoders/autoencoder_dc.py +620 -0
- diffusers/models/autoencoders/autoencoder_kl.py +110 -28
- diffusers/models/autoencoders/autoencoder_kl_allegro.py +1149 -0
- diffusers/models/autoencoders/autoencoder_kl_cogvideox.py +1482 -0
- diffusers/models/autoencoders/autoencoder_kl_hunyuan_video.py +1176 -0
- diffusers/models/autoencoders/autoencoder_kl_ltx.py +1338 -0
- diffusers/models/autoencoders/autoencoder_kl_mochi.py +1166 -0
- diffusers/models/autoencoders/autoencoder_kl_temporal_decoder.py +19 -24
- diffusers/models/autoencoders/autoencoder_oobleck.py +464 -0
- diffusers/models/autoencoders/autoencoder_tiny.py +21 -18
- diffusers/models/autoencoders/consistency_decoder_vae.py +45 -20
- diffusers/models/autoencoders/vae.py +41 -29
- diffusers/models/autoencoders/vq_model.py +182 -0
- diffusers/models/controlnet.py +47 -800
- diffusers/models/controlnet_flux.py +70 -0
- diffusers/models/controlnet_sd3.py +68 -0
- diffusers/models/controlnet_sparsectrl.py +116 -0
- diffusers/models/controlnets/__init__.py +23 -0
- diffusers/models/controlnets/controlnet.py +872 -0
- diffusers/models/{controlnet_flax.py → controlnets/controlnet_flax.py} +9 -9
- diffusers/models/controlnets/controlnet_flux.py +536 -0
- diffusers/models/controlnets/controlnet_hunyuan.py +401 -0
- diffusers/models/controlnets/controlnet_sd3.py +489 -0
- diffusers/models/controlnets/controlnet_sparsectrl.py +788 -0
- diffusers/models/controlnets/controlnet_union.py +832 -0
- diffusers/models/controlnets/controlnet_xs.py +1946 -0
- diffusers/models/controlnets/multicontrolnet.py +183 -0
- diffusers/models/downsampling.py +85 -18
- diffusers/models/embeddings.py +1856 -158
- diffusers/models/embeddings_flax.py +23 -9
- diffusers/models/model_loading_utils.py +480 -0
- diffusers/models/modeling_flax_pytorch_utils.py +2 -1
- diffusers/models/modeling_flax_utils.py +2 -7
- diffusers/models/modeling_outputs.py +14 -0
- diffusers/models/modeling_pytorch_flax_utils.py +1 -1
- diffusers/models/modeling_utils.py +611 -146
- diffusers/models/normalization.py +361 -20
- diffusers/models/resnet.py +18 -23
- diffusers/models/transformers/__init__.py +16 -0
- diffusers/models/transformers/auraflow_transformer_2d.py +544 -0
- diffusers/models/transformers/cogvideox_transformer_3d.py +542 -0
- diffusers/models/transformers/dit_transformer_2d.py +240 -0
- diffusers/models/transformers/dual_transformer_2d.py +9 -8
- diffusers/models/transformers/hunyuan_transformer_2d.py +578 -0
- diffusers/models/transformers/latte_transformer_3d.py +327 -0
- diffusers/models/transformers/lumina_nextdit2d.py +340 -0
- diffusers/models/transformers/pixart_transformer_2d.py +445 -0
- diffusers/models/transformers/prior_transformer.py +13 -13
- diffusers/models/transformers/sana_transformer.py +488 -0
- diffusers/models/transformers/stable_audio_transformer.py +458 -0
- diffusers/models/transformers/t5_film_transformer.py +17 -19
- diffusers/models/transformers/transformer_2d.py +297 -187
- diffusers/models/transformers/transformer_allegro.py +422 -0
- diffusers/models/transformers/transformer_cogview3plus.py +386 -0
- diffusers/models/transformers/transformer_flux.py +593 -0
- diffusers/models/transformers/transformer_hunyuan_video.py +791 -0
- diffusers/models/transformers/transformer_ltx.py +469 -0
- diffusers/models/transformers/transformer_mochi.py +499 -0
- diffusers/models/transformers/transformer_sd3.py +461 -0
- diffusers/models/transformers/transformer_temporal.py +21 -19
- diffusers/models/unets/unet_1d.py +8 -8
- diffusers/models/unets/unet_1d_blocks.py +31 -31
- diffusers/models/unets/unet_2d.py +17 -10
- diffusers/models/unets/unet_2d_blocks.py +225 -149
- diffusers/models/unets/unet_2d_condition.py +41 -40
- diffusers/models/unets/unet_2d_condition_flax.py +6 -5
- diffusers/models/unets/unet_3d_blocks.py +192 -1057
- diffusers/models/unets/unet_3d_condition.py +22 -27
- diffusers/models/unets/unet_i2vgen_xl.py +22 -18
- diffusers/models/unets/unet_kandinsky3.py +2 -2
- diffusers/models/unets/unet_motion_model.py +1413 -89
- diffusers/models/unets/unet_spatio_temporal_condition.py +40 -16
- diffusers/models/unets/unet_stable_cascade.py +19 -18
- diffusers/models/unets/uvit_2d.py +2 -2
- diffusers/models/upsampling.py +95 -26
- diffusers/models/vq_model.py +12 -164
- diffusers/optimization.py +1 -1
- diffusers/pipelines/__init__.py +202 -3
- diffusers/pipelines/allegro/__init__.py +48 -0
- diffusers/pipelines/allegro/pipeline_allegro.py +938 -0
- diffusers/pipelines/allegro/pipeline_output.py +23 -0
- diffusers/pipelines/amused/pipeline_amused.py +12 -12
- diffusers/pipelines/amused/pipeline_amused_img2img.py +14 -12
- diffusers/pipelines/amused/pipeline_amused_inpaint.py +13 -11
- diffusers/pipelines/animatediff/__init__.py +8 -0
- diffusers/pipelines/animatediff/pipeline_animatediff.py +122 -109
- diffusers/pipelines/animatediff/pipeline_animatediff_controlnet.py +1106 -0
- diffusers/pipelines/animatediff/pipeline_animatediff_sdxl.py +1288 -0
- diffusers/pipelines/animatediff/pipeline_animatediff_sparsectrl.py +1010 -0
- diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py +236 -180
- diffusers/pipelines/animatediff/pipeline_animatediff_video2video_controlnet.py +1341 -0
- diffusers/pipelines/animatediff/pipeline_output.py +3 -2
- diffusers/pipelines/audioldm/pipeline_audioldm.py +14 -14
- diffusers/pipelines/audioldm2/modeling_audioldm2.py +58 -39
- diffusers/pipelines/audioldm2/pipeline_audioldm2.py +121 -36
- diffusers/pipelines/aura_flow/__init__.py +48 -0
- diffusers/pipelines/aura_flow/pipeline_aura_flow.py +584 -0
- diffusers/pipelines/auto_pipeline.py +196 -28
- diffusers/pipelines/blip_diffusion/blip_image_processing.py +1 -1
- diffusers/pipelines/blip_diffusion/modeling_blip2.py +6 -6
- diffusers/pipelines/blip_diffusion/modeling_ctx_clip.py +1 -1
- diffusers/pipelines/blip_diffusion/pipeline_blip_diffusion.py +2 -2
- diffusers/pipelines/cogvideo/__init__.py +54 -0
- diffusers/pipelines/cogvideo/pipeline_cogvideox.py +772 -0
- diffusers/pipelines/cogvideo/pipeline_cogvideox_fun_control.py +825 -0
- diffusers/pipelines/cogvideo/pipeline_cogvideox_image2video.py +885 -0
- diffusers/pipelines/cogvideo/pipeline_cogvideox_video2video.py +851 -0
- diffusers/pipelines/cogvideo/pipeline_output.py +20 -0
- diffusers/pipelines/cogview3/__init__.py +47 -0
- diffusers/pipelines/cogview3/pipeline_cogview3plus.py +674 -0
- diffusers/pipelines/cogview3/pipeline_output.py +21 -0
- diffusers/pipelines/consistency_models/pipeline_consistency_models.py +6 -6
- diffusers/pipelines/controlnet/__init__.py +86 -80
- diffusers/pipelines/controlnet/multicontrolnet.py +7 -182
- diffusers/pipelines/controlnet/pipeline_controlnet.py +134 -87
- diffusers/pipelines/controlnet/pipeline_controlnet_blip_diffusion.py +2 -2
- diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py +93 -77
- diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py +88 -197
- diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py +136 -90
- diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py +176 -80
- diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py +125 -89
- diffusers/pipelines/controlnet/pipeline_controlnet_union_inpaint_sd_xl.py +1790 -0
- diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl.py +1501 -0
- diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl_img2img.py +1627 -0
- 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 +1060 -0
- diffusers/pipelines/controlnet_sd3/__init__.py +57 -0
- diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet.py +1133 -0
- diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet_inpainting.py +1153 -0
- diffusers/pipelines/controlnet_xs/__init__.py +68 -0
- diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs.py +916 -0
- diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs_sd_xl.py +1111 -0
- diffusers/pipelines/ddpm/pipeline_ddpm.py +2 -2
- diffusers/pipelines/deepfloyd_if/pipeline_if.py +16 -30
- diffusers/pipelines/deepfloyd_if/pipeline_if_img2img.py +20 -35
- diffusers/pipelines/deepfloyd_if/pipeline_if_img2img_superresolution.py +23 -41
- diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting.py +22 -38
- diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting_superresolution.py +25 -41
- diffusers/pipelines/deepfloyd_if/pipeline_if_superresolution.py +19 -34
- diffusers/pipelines/deepfloyd_if/pipeline_output.py +6 -5
- diffusers/pipelines/deepfloyd_if/watermark.py +1 -1
- diffusers/pipelines/deprecated/alt_diffusion/modeling_roberta_series.py +11 -11
- diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py +70 -30
- diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py +48 -25
- diffusers/pipelines/deprecated/repaint/pipeline_repaint.py +2 -2
- diffusers/pipelines/deprecated/spectrogram_diffusion/pipeline_spectrogram_diffusion.py +7 -7
- diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_cycle_diffusion.py +21 -20
- diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_inpaint_legacy.py +27 -29
- diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_model_editing.py +33 -27
- diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_paradigms.py +33 -23
- diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_pix2pix_zero.py +36 -30
- diffusers/pipelines/deprecated/versatile_diffusion/modeling_text_unet.py +102 -69
- diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion.py +13 -13
- diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion_dual_guided.py +10 -5
- diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion_image_variation.py +11 -6
- diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion_text_to_image.py +10 -5
- diffusers/pipelines/deprecated/vq_diffusion/pipeline_vq_diffusion.py +5 -5
- diffusers/pipelines/dit/pipeline_dit.py +7 -4
- diffusers/pipelines/flux/__init__.py +69 -0
- diffusers/pipelines/flux/modeling_flux.py +47 -0
- diffusers/pipelines/flux/pipeline_flux.py +957 -0
- diffusers/pipelines/flux/pipeline_flux_control.py +889 -0
- diffusers/pipelines/flux/pipeline_flux_control_img2img.py +945 -0
- diffusers/pipelines/flux/pipeline_flux_control_inpaint.py +1141 -0
- diffusers/pipelines/flux/pipeline_flux_controlnet.py +1006 -0
- diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py +998 -0
- diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py +1204 -0
- diffusers/pipelines/flux/pipeline_flux_fill.py +969 -0
- diffusers/pipelines/flux/pipeline_flux_img2img.py +856 -0
- diffusers/pipelines/flux/pipeline_flux_inpaint.py +1022 -0
- diffusers/pipelines/flux/pipeline_flux_prior_redux.py +492 -0
- diffusers/pipelines/flux/pipeline_output.py +37 -0
- diffusers/pipelines/free_init_utils.py +41 -38
- diffusers/pipelines/free_noise_utils.py +596 -0
- diffusers/pipelines/hunyuan_video/__init__.py +48 -0
- diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video.py +687 -0
- diffusers/pipelines/hunyuan_video/pipeline_output.py +20 -0
- diffusers/pipelines/hunyuandit/__init__.py +48 -0
- diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py +916 -0
- diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py +33 -48
- diffusers/pipelines/kandinsky/pipeline_kandinsky.py +8 -8
- diffusers/pipelines/kandinsky/pipeline_kandinsky_combined.py +32 -29
- diffusers/pipelines/kandinsky/pipeline_kandinsky_img2img.py +11 -11
- diffusers/pipelines/kandinsky/pipeline_kandinsky_inpaint.py +12 -12
- diffusers/pipelines/kandinsky/pipeline_kandinsky_prior.py +10 -10
- diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2.py +6 -6
- diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_combined.py +34 -31
- diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_controlnet.py +10 -10
- diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_controlnet_img2img.py +10 -10
- diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_img2img.py +6 -6
- diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_inpainting.py +8 -8
- diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior.py +7 -7
- diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior_emb2emb.py +6 -6
- diffusers/pipelines/kandinsky3/convert_kandinsky3_unet.py +3 -3
- diffusers/pipelines/kandinsky3/pipeline_kandinsky3.py +22 -35
- diffusers/pipelines/kandinsky3/pipeline_kandinsky3_img2img.py +26 -37
- diffusers/pipelines/kolors/__init__.py +54 -0
- diffusers/pipelines/kolors/pipeline_kolors.py +1070 -0
- diffusers/pipelines/kolors/pipeline_kolors_img2img.py +1250 -0
- diffusers/pipelines/kolors/pipeline_output.py +21 -0
- diffusers/pipelines/kolors/text_encoder.py +889 -0
- diffusers/pipelines/kolors/tokenizer.py +338 -0
- diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_img2img.py +82 -62
- diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_text2img.py +77 -60
- diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py +12 -12
- diffusers/pipelines/latte/__init__.py +48 -0
- diffusers/pipelines/latte/pipeline_latte.py +881 -0
- diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py +80 -74
- diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py +85 -76
- diffusers/pipelines/ledits_pp/pipeline_output.py +2 -2
- diffusers/pipelines/ltx/__init__.py +50 -0
- diffusers/pipelines/ltx/pipeline_ltx.py +789 -0
- diffusers/pipelines/ltx/pipeline_ltx_image2video.py +885 -0
- diffusers/pipelines/ltx/pipeline_output.py +20 -0
- diffusers/pipelines/lumina/__init__.py +48 -0
- diffusers/pipelines/lumina/pipeline_lumina.py +890 -0
- diffusers/pipelines/marigold/__init__.py +50 -0
- diffusers/pipelines/marigold/marigold_image_processing.py +576 -0
- diffusers/pipelines/marigold/pipeline_marigold_depth.py +813 -0
- diffusers/pipelines/marigold/pipeline_marigold_normals.py +690 -0
- diffusers/pipelines/mochi/__init__.py +48 -0
- diffusers/pipelines/mochi/pipeline_mochi.py +748 -0
- diffusers/pipelines/mochi/pipeline_output.py +20 -0
- diffusers/pipelines/musicldm/pipeline_musicldm.py +14 -14
- diffusers/pipelines/pag/__init__.py +80 -0
- diffusers/pipelines/pag/pag_utils.py +243 -0
- diffusers/pipelines/pag/pipeline_pag_controlnet_sd.py +1328 -0
- diffusers/pipelines/pag/pipeline_pag_controlnet_sd_inpaint.py +1543 -0
- diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl.py +1610 -0
- diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl_img2img.py +1683 -0
- diffusers/pipelines/pag/pipeline_pag_hunyuandit.py +969 -0
- diffusers/pipelines/pag/pipeline_pag_kolors.py +1136 -0
- diffusers/pipelines/pag/pipeline_pag_pixart_sigma.py +865 -0
- diffusers/pipelines/pag/pipeline_pag_sana.py +886 -0
- diffusers/pipelines/pag/pipeline_pag_sd.py +1062 -0
- diffusers/pipelines/pag/pipeline_pag_sd_3.py +994 -0
- diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py +1058 -0
- diffusers/pipelines/pag/pipeline_pag_sd_animatediff.py +866 -0
- diffusers/pipelines/pag/pipeline_pag_sd_img2img.py +1094 -0
- diffusers/pipelines/pag/pipeline_pag_sd_inpaint.py +1356 -0
- diffusers/pipelines/pag/pipeline_pag_sd_xl.py +1345 -0
- diffusers/pipelines/pag/pipeline_pag_sd_xl_img2img.py +1544 -0
- diffusers/pipelines/pag/pipeline_pag_sd_xl_inpaint.py +1776 -0
- diffusers/pipelines/paint_by_example/pipeline_paint_by_example.py +17 -12
- diffusers/pipelines/pia/pipeline_pia.py +74 -164
- diffusers/pipelines/pipeline_flax_utils.py +5 -10
- diffusers/pipelines/pipeline_loading_utils.py +515 -53
- diffusers/pipelines/pipeline_utils.py +411 -222
- diffusers/pipelines/pixart_alpha/__init__.py +8 -1
- diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py +76 -93
- diffusers/pipelines/pixart_alpha/pipeline_pixart_sigma.py +873 -0
- diffusers/pipelines/sana/__init__.py +47 -0
- diffusers/pipelines/sana/pipeline_output.py +21 -0
- diffusers/pipelines/sana/pipeline_sana.py +884 -0
- diffusers/pipelines/semantic_stable_diffusion/pipeline_semantic_stable_diffusion.py +27 -23
- diffusers/pipelines/shap_e/pipeline_shap_e.py +3 -3
- diffusers/pipelines/shap_e/pipeline_shap_e_img2img.py +14 -14
- diffusers/pipelines/shap_e/renderer.py +1 -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 +756 -0
- diffusers/pipelines/stable_cascade/pipeline_stable_cascade.py +71 -25
- diffusers/pipelines/stable_cascade/pipeline_stable_cascade_combined.py +23 -19
- diffusers/pipelines/stable_cascade/pipeline_stable_cascade_prior.py +35 -34
- diffusers/pipelines/stable_diffusion/__init__.py +0 -1
- diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +20 -11
- diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion.py +1 -1
- diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py +2 -2
- diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_upscale.py +6 -6
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py +145 -79
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py +43 -28
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_image_variation.py +13 -8
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py +100 -68
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py +109 -201
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py +131 -32
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_latent_upscale.py +247 -87
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py +30 -29
- diffusers/pipelines/stable_diffusion/pipeline_stable_unclip.py +35 -27
- diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py +49 -42
- diffusers/pipelines/stable_diffusion/safety_checker.py +2 -1
- diffusers/pipelines/stable_diffusion_3/__init__.py +54 -0
- diffusers/pipelines/stable_diffusion_3/pipeline_output.py +21 -0
- diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py +1140 -0
- diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py +1036 -0
- diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py +1250 -0
- diffusers/pipelines/stable_diffusion_attend_and_excite/pipeline_stable_diffusion_attend_and_excite.py +29 -20
- diffusers/pipelines/stable_diffusion_diffedit/pipeline_stable_diffusion_diffedit.py +59 -58
- diffusers/pipelines/stable_diffusion_gligen/pipeline_stable_diffusion_gligen.py +31 -25
- diffusers/pipelines/stable_diffusion_gligen/pipeline_stable_diffusion_gligen_text_image.py +38 -22
- diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_k_diffusion.py +30 -24
- diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py +24 -23
- diffusers/pipelines/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py +107 -67
- diffusers/pipelines/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py +316 -69
- diffusers/pipelines/stable_diffusion_safe/pipeline_stable_diffusion_safe.py +10 -5
- diffusers/pipelines/stable_diffusion_safe/safety_checker.py +1 -1
- diffusers/pipelines/stable_diffusion_sag/pipeline_stable_diffusion_sag.py +98 -30
- diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py +121 -83
- diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py +161 -105
- diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py +142 -218
- diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_instruct_pix2pix.py +45 -29
- diffusers/pipelines/stable_diffusion_xl/watermark.py +9 -3
- diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py +110 -57
- diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_adapter.py +69 -39
- diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_xl_adapter.py +105 -74
- diffusers/pipelines/text_to_video_synthesis/pipeline_output.py +3 -2
- diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth.py +29 -49
- diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth_img2img.py +32 -93
- diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero.py +37 -25
- diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero_sdxl.py +54 -40
- diffusers/pipelines/unclip/pipeline_unclip.py +6 -6
- diffusers/pipelines/unclip/pipeline_unclip_image_variation.py +6 -6
- diffusers/pipelines/unidiffuser/modeling_text_decoder.py +1 -1
- diffusers/pipelines/unidiffuser/modeling_uvit.py +12 -12
- diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py +29 -28
- diffusers/pipelines/wuerstchen/modeling_paella_vq_model.py +5 -5
- diffusers/pipelines/wuerstchen/modeling_wuerstchen_common.py +5 -10
- diffusers/pipelines/wuerstchen/modeling_wuerstchen_prior.py +6 -8
- diffusers/pipelines/wuerstchen/pipeline_wuerstchen.py +4 -4
- diffusers/pipelines/wuerstchen/pipeline_wuerstchen_combined.py +12 -12
- diffusers/pipelines/wuerstchen/pipeline_wuerstchen_prior.py +15 -14
- diffusers/{models/dual_transformer_2d.py → quantizers/__init__.py} +2 -6
- diffusers/quantizers/auto.py +139 -0
- diffusers/quantizers/base.py +233 -0
- diffusers/quantizers/bitsandbytes/__init__.py +2 -0
- diffusers/quantizers/bitsandbytes/bnb_quantizer.py +561 -0
- diffusers/quantizers/bitsandbytes/utils.py +306 -0
- diffusers/quantizers/gguf/__init__.py +1 -0
- diffusers/quantizers/gguf/gguf_quantizer.py +159 -0
- diffusers/quantizers/gguf/utils.py +456 -0
- diffusers/quantizers/quantization_config.py +669 -0
- diffusers/quantizers/torchao/__init__.py +15 -0
- diffusers/quantizers/torchao/torchao_quantizer.py +292 -0
- diffusers/schedulers/__init__.py +12 -2
- diffusers/schedulers/deprecated/__init__.py +1 -1
- diffusers/schedulers/deprecated/scheduling_karras_ve.py +25 -25
- diffusers/schedulers/scheduling_amused.py +5 -5
- diffusers/schedulers/scheduling_consistency_decoder.py +11 -11
- diffusers/schedulers/scheduling_consistency_models.py +23 -25
- diffusers/schedulers/scheduling_cosine_dpmsolver_multistep.py +572 -0
- diffusers/schedulers/scheduling_ddim.py +27 -26
- diffusers/schedulers/scheduling_ddim_cogvideox.py +452 -0
- diffusers/schedulers/scheduling_ddim_flax.py +2 -1
- diffusers/schedulers/scheduling_ddim_inverse.py +16 -16
- diffusers/schedulers/scheduling_ddim_parallel.py +32 -31
- diffusers/schedulers/scheduling_ddpm.py +27 -30
- diffusers/schedulers/scheduling_ddpm_flax.py +7 -3
- diffusers/schedulers/scheduling_ddpm_parallel.py +33 -36
- diffusers/schedulers/scheduling_ddpm_wuerstchen.py +14 -14
- diffusers/schedulers/scheduling_deis_multistep.py +150 -50
- diffusers/schedulers/scheduling_dpm_cogvideox.py +489 -0
- diffusers/schedulers/scheduling_dpmsolver_multistep.py +221 -84
- diffusers/schedulers/scheduling_dpmsolver_multistep_flax.py +2 -2
- diffusers/schedulers/scheduling_dpmsolver_multistep_inverse.py +158 -52
- diffusers/schedulers/scheduling_dpmsolver_sde.py +153 -34
- diffusers/schedulers/scheduling_dpmsolver_singlestep.py +275 -86
- diffusers/schedulers/scheduling_edm_dpmsolver_multistep.py +81 -57
- diffusers/schedulers/scheduling_edm_euler.py +62 -39
- diffusers/schedulers/scheduling_euler_ancestral_discrete.py +30 -29
- diffusers/schedulers/scheduling_euler_discrete.py +255 -74
- diffusers/schedulers/scheduling_flow_match_euler_discrete.py +458 -0
- diffusers/schedulers/scheduling_flow_match_heun_discrete.py +320 -0
- diffusers/schedulers/scheduling_heun_discrete.py +174 -46
- diffusers/schedulers/scheduling_ipndm.py +9 -9
- diffusers/schedulers/scheduling_k_dpm_2_ancestral_discrete.py +138 -29
- diffusers/schedulers/scheduling_k_dpm_2_discrete.py +132 -26
- diffusers/schedulers/scheduling_karras_ve_flax.py +6 -6
- diffusers/schedulers/scheduling_lcm.py +23 -29
- diffusers/schedulers/scheduling_lms_discrete.py +105 -28
- diffusers/schedulers/scheduling_pndm.py +20 -20
- diffusers/schedulers/scheduling_repaint.py +21 -21
- diffusers/schedulers/scheduling_sasolver.py +157 -60
- diffusers/schedulers/scheduling_sde_ve.py +19 -19
- diffusers/schedulers/scheduling_tcd.py +41 -36
- diffusers/schedulers/scheduling_unclip.py +19 -16
- diffusers/schedulers/scheduling_unipc_multistep.py +243 -47
- diffusers/schedulers/scheduling_utils.py +12 -5
- diffusers/schedulers/scheduling_utils_flax.py +1 -3
- diffusers/schedulers/scheduling_vq_diffusion.py +10 -10
- diffusers/training_utils.py +214 -30
- diffusers/utils/__init__.py +17 -1
- diffusers/utils/constants.py +3 -0
- diffusers/utils/doc_utils.py +1 -0
- diffusers/utils/dummy_pt_objects.py +592 -7
- 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 +1001 -71
- diffusers/utils/dynamic_modules_utils.py +34 -29
- diffusers/utils/export_utils.py +50 -6
- diffusers/utils/hub_utils.py +131 -17
- diffusers/utils/import_utils.py +210 -8
- diffusers/utils/loading_utils.py +118 -5
- diffusers/utils/logging.py +4 -2
- diffusers/utils/peft_utils.py +37 -7
- diffusers/utils/state_dict_utils.py +13 -2
- diffusers/utils/testing_utils.py +193 -11
- diffusers/utils/torch_utils.py +4 -0
- diffusers/video_processor.py +113 -0
- {diffusers-0.27.1.dist-info → diffusers-0.32.2.dist-info}/METADATA +82 -91
- diffusers-0.32.2.dist-info/RECORD +550 -0
- {diffusers-0.27.1.dist-info → diffusers-0.32.2.dist-info}/WHEEL +1 -1
- diffusers/loaders/autoencoder.py +0 -146
- diffusers/loaders/controlnet.py +0 -136
- diffusers/loaders/lora.py +0 -1349
- diffusers/models/prior_transformer.py +0 -12
- diffusers/models/t5_film_transformer.py +0 -70
- diffusers/models/transformer_2d.py +0 -25
- diffusers/models/transformer_temporal.py +0 -34
- diffusers/models/unet_1d.py +0 -26
- diffusers/models/unet_1d_blocks.py +0 -203
- diffusers/models/unet_2d.py +0 -27
- diffusers/models/unet_2d_blocks.py +0 -375
- diffusers/models/unet_2d_condition.py +0 -25
- diffusers-0.27.1.dist-info/RECORD +0 -399
- {diffusers-0.27.1.dist-info → diffusers-0.32.2.dist-info}/LICENSE +0 -0
- {diffusers-0.27.1.dist-info → diffusers-0.32.2.dist-info}/entry_points.txt +0 -0
- {diffusers-0.27.1.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
|
-
|
16
|
-
|
17
|
-
from
|
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
|
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 `
|
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
|
-
|
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)
|