diffusers 0.27.0__tar.gz → 0.32.2__tar.gz
Sign up to get free protection for your applications and to get access to all the features.
- diffusers-0.32.2/PKG-INFO +344 -0
- diffusers-0.32.2/README.md +239 -0
- diffusers-0.32.2/pyproject.toml +29 -0
- diffusers-0.32.2/setup.py +286 -0
- diffusers-0.32.2/src/diffusers/__init__.py +1014 -0
- diffusers-0.32.2/src/diffusers/callbacks.py +209 -0
- diffusers-0.32.2/src/diffusers/commands/env.py +180 -0
- diffusers-0.32.2/src/diffusers/configuration_utils.py +732 -0
- diffusers-0.32.2/src/diffusers/dependency_versions_table.py +46 -0
- diffusers-0.32.2/src/diffusers/image_processor.py +1314 -0
- diffusers-0.32.2/src/diffusers/loaders/__init__.py +121 -0
- diffusers-0.32.2/src/diffusers/loaders/ip_adapter.py +871 -0
- diffusers-0.32.2/src/diffusers/loaders/lora_base.py +900 -0
- diffusers-0.32.2/src/diffusers/loaders/lora_conversion_utils.py +1150 -0
- diffusers-0.32.2/src/diffusers/loaders/lora_pipeline.py +3812 -0
- diffusers-0.32.2/src/diffusers/loaders/peft.py +750 -0
- diffusers-0.32.2/src/diffusers/loaders/single_file.py +550 -0
- diffusers-0.32.2/src/diffusers/loaders/single_file_model.py +385 -0
- diffusers-0.32.2/src/diffusers/loaders/single_file_utils.py +2687 -0
- diffusers-0.32.2/src/diffusers/loaders/textual_inversion.py +580 -0
- diffusers-0.32.2/src/diffusers/loaders/transformer_flux.py +181 -0
- diffusers-0.32.2/src/diffusers/loaders/transformer_sd3.py +89 -0
- diffusers-0.32.2/src/diffusers/loaders/unet.py +927 -0
- diffusers-0.32.2/src/diffusers/loaders/unet_loader_utils.py +163 -0
- diffusers-0.32.2/src/diffusers/models/__init__.py +172 -0
- diffusers-0.32.2/src/diffusers/models/activations.py +178 -0
- diffusers-0.32.2/src/diffusers/models/adapter.py +584 -0
- diffusers-0.32.2/src/diffusers/models/attention.py +1252 -0
- diffusers-0.32.2/src/diffusers/models/attention_flax.py +494 -0
- diffusers-0.32.2/src/diffusers/models/attention_processor.py +6097 -0
- diffusers-0.32.2/src/diffusers/models/autoencoders/__init__.py +13 -0
- diffusers-0.32.2/src/diffusers/models/autoencoders/autoencoder_asym_kl.py +184 -0
- diffusers-0.32.2/src/diffusers/models/autoencoders/autoencoder_dc.py +620 -0
- diffusers-0.32.2/src/diffusers/models/autoencoders/autoencoder_kl.py +571 -0
- diffusers-0.32.2/src/diffusers/models/autoencoders/autoencoder_kl_allegro.py +1149 -0
- diffusers-0.32.2/src/diffusers/models/autoencoders/autoencoder_kl_cogvideox.py +1482 -0
- diffusers-0.32.2/src/diffusers/models/autoencoders/autoencoder_kl_hunyuan_video.py +1176 -0
- diffusers-0.32.2/src/diffusers/models/autoencoders/autoencoder_kl_ltx.py +1338 -0
- diffusers-0.32.2/src/diffusers/models/autoencoders/autoencoder_kl_mochi.py +1166 -0
- diffusers-0.32.2/src/diffusers/models/autoencoders/autoencoder_kl_temporal_decoder.py +394 -0
- diffusers-0.32.2/src/diffusers/models/autoencoders/autoencoder_oobleck.py +464 -0
- diffusers-0.32.2/src/diffusers/models/autoencoders/autoencoder_tiny.py +350 -0
- diffusers-0.32.2/src/diffusers/models/autoencoders/consistency_decoder_vae.py +460 -0
- diffusers-0.32.2/src/diffusers/models/autoencoders/vae.py +995 -0
- diffusers-0.32.2/src/diffusers/models/autoencoders/vq_model.py +182 -0
- diffusers-0.32.2/src/diffusers/models/controlnet.py +115 -0
- diffusers-0.32.2/src/diffusers/models/controlnet_flux.py +70 -0
- diffusers-0.32.2/src/diffusers/models/controlnet_sd3.py +68 -0
- diffusers-0.32.2/src/diffusers/models/controlnet_sparsectrl.py +116 -0
- diffusers-0.32.2/src/diffusers/models/controlnets/__init__.py +23 -0
- diffusers-0.32.2/src/diffusers/models/controlnets/controlnet.py +872 -0
- diffusers-0.32.2/src/diffusers/models/controlnets/controlnet_flax.py +395 -0
- diffusers-0.32.2/src/diffusers/models/controlnets/controlnet_flux.py +536 -0
- diffusers-0.32.2/src/diffusers/models/controlnets/controlnet_hunyuan.py +401 -0
- diffusers-0.32.2/src/diffusers/models/controlnets/controlnet_sd3.py +489 -0
- diffusers-0.32.2/src/diffusers/models/controlnets/controlnet_sparsectrl.py +788 -0
- diffusers-0.32.2/src/diffusers/models/controlnets/controlnet_union.py +832 -0
- diffusers-0.32.2/src/diffusers/models/controlnets/controlnet_xs.py +1946 -0
- diffusers-0.32.2/src/diffusers/models/controlnets/multicontrolnet.py +183 -0
- diffusers-0.32.2/src/diffusers/models/downsampling.py +401 -0
- diffusers-0.32.2/src/diffusers/models/embeddings.py +2612 -0
- diffusers-0.32.2/src/diffusers/models/embeddings_flax.py +111 -0
- diffusers-0.32.2/src/diffusers/models/model_loading_utils.py +480 -0
- diffusers-0.32.2/src/diffusers/models/modeling_flax_pytorch_utils.py +135 -0
- diffusers-0.32.2/src/diffusers/models/modeling_flax_utils.py +561 -0
- diffusers-0.32.2/src/diffusers/models/modeling_outputs.py +31 -0
- diffusers-0.32.2/src/diffusers/models/modeling_pytorch_flax_utils.py +161 -0
- diffusers-0.32.2/src/diffusers/models/modeling_utils.py +1486 -0
- diffusers-0.32.2/src/diffusers/models/normalization.py +595 -0
- diffusers-0.32.2/src/diffusers/models/resnet.py +797 -0
- diffusers-0.32.2/src/diffusers/models/transformers/__init__.py +25 -0
- diffusers-0.32.2/src/diffusers/models/transformers/auraflow_transformer_2d.py +544 -0
- diffusers-0.32.2/src/diffusers/models/transformers/cogvideox_transformer_3d.py +542 -0
- diffusers-0.32.2/src/diffusers/models/transformers/dit_transformer_2d.py +240 -0
- diffusers-0.32.2/src/diffusers/models/transformers/dual_transformer_2d.py +156 -0
- diffusers-0.32.2/src/diffusers/models/transformers/hunyuan_transformer_2d.py +578 -0
- diffusers-0.32.2/src/diffusers/models/transformers/latte_transformer_3d.py +327 -0
- diffusers-0.32.2/src/diffusers/models/transformers/lumina_nextdit2d.py +340 -0
- diffusers-0.32.2/src/diffusers/models/transformers/pixart_transformer_2d.py +445 -0
- diffusers-0.32.2/src/diffusers/models/transformers/prior_transformer.py +380 -0
- diffusers-0.32.2/src/diffusers/models/transformers/sana_transformer.py +488 -0
- diffusers-0.32.2/src/diffusers/models/transformers/stable_audio_transformer.py +458 -0
- diffusers-0.32.2/src/diffusers/models/transformers/t5_film_transformer.py +436 -0
- diffusers-0.32.2/src/diffusers/models/transformers/transformer_2d.py +566 -0
- diffusers-0.32.2/src/diffusers/models/transformers/transformer_allegro.py +422 -0
- diffusers-0.32.2/src/diffusers/models/transformers/transformer_cogview3plus.py +386 -0
- diffusers-0.32.2/src/diffusers/models/transformers/transformer_flux.py +593 -0
- diffusers-0.32.2/src/diffusers/models/transformers/transformer_hunyuan_video.py +791 -0
- diffusers-0.32.2/src/diffusers/models/transformers/transformer_ltx.py +469 -0
- diffusers-0.32.2/src/diffusers/models/transformers/transformer_mochi.py +499 -0
- diffusers-0.32.2/src/diffusers/models/transformers/transformer_sd3.py +461 -0
- diffusers-0.32.2/src/diffusers/models/transformers/transformer_temporal.py +381 -0
- diffusers-0.32.2/src/diffusers/models/unets/unet_1d.py +255 -0
- diffusers-0.32.2/src/diffusers/models/unets/unet_1d_blocks.py +702 -0
- diffusers-0.32.2/src/diffusers/models/unets/unet_2d.py +353 -0
- diffusers-0.32.2/src/diffusers/models/unets/unet_2d_blocks.py +3807 -0
- diffusers-0.32.2/src/diffusers/models/unets/unet_2d_condition.py +1312 -0
- diffusers-0.32.2/src/diffusers/models/unets/unet_2d_condition_flax.py +454 -0
- diffusers-0.32.2/src/diffusers/models/unets/unet_3d_blocks.py +1540 -0
- diffusers-0.32.2/src/diffusers/models/unets/unet_3d_condition.py +748 -0
- diffusers-0.32.2/src/diffusers/models/unets/unet_i2vgen_xl.py +728 -0
- diffusers-0.32.2/src/diffusers/models/unets/unet_kandinsky3.py +535 -0
- diffusers-0.32.2/src/diffusers/models/unets/unet_motion_model.py +2272 -0
- diffusers-0.32.2/src/diffusers/models/unets/unet_spatio_temporal_condition.py +513 -0
- diffusers-0.32.2/src/diffusers/models/unets/unet_stable_cascade.py +611 -0
- diffusers-0.32.2/src/diffusers/models/unets/uvit_2d.py +470 -0
- diffusers-0.32.2/src/diffusers/models/upsampling.py +517 -0
- diffusers-0.32.2/src/diffusers/models/vq_model.py +29 -0
- diffusers-0.32.2/src/diffusers/optimization.py +361 -0
- diffusers-0.32.2/src/diffusers/pipelines/__init__.py +780 -0
- diffusers-0.32.2/src/diffusers/pipelines/allegro/__init__.py +48 -0
- diffusers-0.32.2/src/diffusers/pipelines/allegro/pipeline_allegro.py +938 -0
- diffusers-0.32.2/src/diffusers/pipelines/allegro/pipeline_output.py +23 -0
- diffusers-0.32.2/src/diffusers/pipelines/amused/pipeline_amused.py +328 -0
- diffusers-0.32.2/src/diffusers/pipelines/amused/pipeline_amused_img2img.py +349 -0
- diffusers-0.32.2/src/diffusers/pipelines/amused/pipeline_amused_inpaint.py +380 -0
- diffusers-0.32.2/src/diffusers/pipelines/animatediff/__init__.py +57 -0
- diffusers-0.32.2/src/diffusers/pipelines/animatediff/pipeline_animatediff.py +860 -0
- diffusers-0.32.2/src/diffusers/pipelines/animatediff/pipeline_animatediff_controlnet.py +1106 -0
- diffusers-0.32.2/src/diffusers/pipelines/animatediff/pipeline_animatediff_sdxl.py +1288 -0
- diffusers-0.32.2/src/diffusers/pipelines/animatediff/pipeline_animatediff_sparsectrl.py +1010 -0
- diffusers-0.32.2/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py +1053 -0
- diffusers-0.32.2/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video_controlnet.py +1341 -0
- diffusers-0.32.2/src/diffusers/pipelines/animatediff/pipeline_output.py +24 -0
- diffusers-0.32.2/src/diffusers/pipelines/audioldm/pipeline_audioldm.py +546 -0
- diffusers-0.32.2/src/diffusers/pipelines/audioldm2/modeling_audioldm2.py +1530 -0
- diffusers-0.32.2/src/diffusers/pipelines/audioldm2/pipeline_audioldm2.py +1065 -0
- diffusers-0.32.2/src/diffusers/pipelines/aura_flow/__init__.py +48 -0
- diffusers-0.32.2/src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py +584 -0
- diffusers-0.32.2/src/diffusers/pipelines/auto_pipeline.py +1155 -0
- diffusers-0.32.2/src/diffusers/pipelines/blip_diffusion/blip_image_processing.py +318 -0
- diffusers-0.32.2/src/diffusers/pipelines/blip_diffusion/modeling_blip2.py +642 -0
- diffusers-0.32.2/src/diffusers/pipelines/blip_diffusion/modeling_ctx_clip.py +223 -0
- diffusers-0.32.2/src/diffusers/pipelines/blip_diffusion/pipeline_blip_diffusion.py +348 -0
- diffusers-0.32.2/src/diffusers/pipelines/cogvideo/__init__.py +54 -0
- diffusers-0.32.2/src/diffusers/pipelines/cogvideo/pipeline_cogvideox.py +772 -0
- diffusers-0.32.2/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_fun_control.py +825 -0
- diffusers-0.32.2/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_image2video.py +885 -0
- diffusers-0.32.2/src/diffusers/pipelines/cogvideo/pipeline_cogvideox_video2video.py +851 -0
- diffusers-0.32.2/src/diffusers/pipelines/cogvideo/pipeline_output.py +20 -0
- diffusers-0.32.2/src/diffusers/pipelines/cogview3/__init__.py +47 -0
- diffusers-0.32.2/src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py +674 -0
- diffusers-0.32.2/src/diffusers/pipelines/cogview3/pipeline_output.py +21 -0
- diffusers-0.32.2/src/diffusers/pipelines/consistency_models/pipeline_consistency_models.py +275 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet/__init__.py +86 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet/multicontrolnet.py +12 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet/pipeline_controlnet.py +1365 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet/pipeline_controlnet_blip_diffusion.py +413 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py +1326 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py +1511 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py +1864 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py +1595 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py +1662 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_inpaint_sd_xl.py +1790 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl.py +1501 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl_img2img.py +1627 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet/pipeline_flax_controlnet.py +532 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet_hunyuandit/__init__.py +48 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py +1060 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet_sd3/__init__.py +57 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet.py +1133 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet_inpainting.py +1153 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet_xs/__init__.py +68 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs.py +916 -0
- diffusers-0.32.2/src/diffusers/pipelines/controlnet_xs/pipeline_controlnet_xs_sd_xl.py +1111 -0
- diffusers-0.32.2/src/diffusers/pipelines/ddpm/pipeline_ddpm.py +127 -0
- diffusers-0.32.2/src/diffusers/pipelines/deepfloyd_if/pipeline_if.py +774 -0
- diffusers-0.32.2/src/diffusers/pipelines/deepfloyd_if/pipeline_if_img2img.py +895 -0
- diffusers-0.32.2/src/diffusers/pipelines/deepfloyd_if/pipeline_if_img2img_superresolution.py +1011 -0
- diffusers-0.32.2/src/diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting.py +1014 -0
- diffusers-0.32.2/src/diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting_superresolution.py +1121 -0
- diffusers-0.32.2/src/diffusers/pipelines/deepfloyd_if/pipeline_if_superresolution.py +870 -0
- diffusers-0.32.2/src/diffusers/pipelines/deepfloyd_if/pipeline_output.py +29 -0
- diffusers-0.32.2/src/diffusers/pipelines/deepfloyd_if/watermark.py +46 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/alt_diffusion/modeling_roberta_series.py +124 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py +986 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py +1041 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/repaint/pipeline_repaint.py +230 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/spectrogram_diffusion/pipeline_spectrogram_diffusion.py +269 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_cycle_diffusion.py +949 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_inpaint_legacy.py +784 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_model_editing.py +830 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_paradigms.py +796 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_pix2pix_zero.py +1310 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/versatile_diffusion/modeling_text_unet.py +2541 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion.py +421 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion_dual_guided.py +561 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion_image_variation.py +402 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion_text_to_image.py +480 -0
- diffusers-0.32.2/src/diffusers/pipelines/deprecated/vq_diffusion/pipeline_vq_diffusion.py +325 -0
- diffusers-0.32.2/src/diffusers/pipelines/dit/pipeline_dit.py +236 -0
- diffusers-0.32.2/src/diffusers/pipelines/flux/__init__.py +69 -0
- diffusers-0.32.2/src/diffusers/pipelines/flux/modeling_flux.py +47 -0
- diffusers-0.32.2/src/diffusers/pipelines/flux/pipeline_flux.py +957 -0
- diffusers-0.32.2/src/diffusers/pipelines/flux/pipeline_flux_control.py +889 -0
- diffusers-0.32.2/src/diffusers/pipelines/flux/pipeline_flux_control_img2img.py +945 -0
- diffusers-0.32.2/src/diffusers/pipelines/flux/pipeline_flux_control_inpaint.py +1141 -0
- diffusers-0.32.2/src/diffusers/pipelines/flux/pipeline_flux_controlnet.py +1006 -0
- diffusers-0.32.2/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py +998 -0
- diffusers-0.32.2/src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py +1204 -0
- diffusers-0.32.2/src/diffusers/pipelines/flux/pipeline_flux_fill.py +969 -0
- diffusers-0.32.2/src/diffusers/pipelines/flux/pipeline_flux_img2img.py +856 -0
- diffusers-0.32.2/src/diffusers/pipelines/flux/pipeline_flux_inpaint.py +1022 -0
- diffusers-0.32.2/src/diffusers/pipelines/flux/pipeline_flux_prior_redux.py +492 -0
- diffusers-0.32.2/src/diffusers/pipelines/flux/pipeline_output.py +37 -0
- diffusers-0.32.2/src/diffusers/pipelines/free_init_utils.py +187 -0
- diffusers-0.32.2/src/diffusers/pipelines/free_noise_utils.py +596 -0
- diffusers-0.32.2/src/diffusers/pipelines/hunyuan_video/__init__.py +48 -0
- diffusers-0.32.2/src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video.py +687 -0
- diffusers-0.32.2/src/diffusers/pipelines/hunyuan_video/pipeline_output.py +20 -0
- diffusers-0.32.2/src/diffusers/pipelines/hunyuandit/__init__.py +48 -0
- diffusers-0.32.2/src/diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py +916 -0
- diffusers-0.32.2/src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py +783 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky/pipeline_kandinsky.py +407 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky/pipeline_kandinsky_combined.py +817 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky/pipeline_kandinsky_img2img.py +500 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky/pipeline_kandinsky_inpaint.py +635 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky/pipeline_kandinsky_prior.py +547 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2.py +320 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_combined.py +854 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_controlnet.py +320 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_controlnet_img2img.py +381 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_img2img.py +399 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_inpainting.py +556 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior.py +549 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior_emb2emb.py +563 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky3/convert_kandinsky3_unet.py +98 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3.py +576 -0
- diffusers-0.32.2/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3_img2img.py +643 -0
- diffusers-0.32.2/src/diffusers/pipelines/kolors/__init__.py +54 -0
- diffusers-0.32.2/src/diffusers/pipelines/kolors/pipeline_kolors.py +1070 -0
- diffusers-0.32.2/src/diffusers/pipelines/kolors/pipeline_kolors_img2img.py +1250 -0
- diffusers-0.32.2/src/diffusers/pipelines/kolors/pipeline_output.py +21 -0
- diffusers-0.32.2/src/diffusers/pipelines/kolors/text_encoder.py +889 -0
- diffusers-0.32.2/src/diffusers/pipelines/kolors/tokenizer.py +338 -0
- diffusers-0.32.2/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_img2img.py +976 -0
- diffusers-0.32.2/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_text2img.py +905 -0
- diffusers-0.32.2/src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py +746 -0
- diffusers-0.32.2/src/diffusers/pipelines/latte/__init__.py +48 -0
- diffusers-0.32.2/src/diffusers/pipelines/latte/pipeline_latte.py +881 -0
- diffusers-0.32.2/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py +1511 -0
- diffusers-0.32.2/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py +1806 -0
- diffusers-0.32.2/src/diffusers/pipelines/ledits_pp/pipeline_output.py +43 -0
- diffusers-0.32.2/src/diffusers/pipelines/ltx/__init__.py +50 -0
- diffusers-0.32.2/src/diffusers/pipelines/ltx/pipeline_ltx.py +789 -0
- diffusers-0.32.2/src/diffusers/pipelines/ltx/pipeline_ltx_image2video.py +885 -0
- diffusers-0.32.2/src/diffusers/pipelines/ltx/pipeline_output.py +20 -0
- diffusers-0.32.2/src/diffusers/pipelines/lumina/__init__.py +48 -0
- diffusers-0.32.2/src/diffusers/pipelines/lumina/pipeline_lumina.py +890 -0
- diffusers-0.32.2/src/diffusers/pipelines/marigold/__init__.py +50 -0
- diffusers-0.32.2/src/diffusers/pipelines/marigold/marigold_image_processing.py +576 -0
- diffusers-0.32.2/src/diffusers/pipelines/marigold/pipeline_marigold_depth.py +813 -0
- diffusers-0.32.2/src/diffusers/pipelines/marigold/pipeline_marigold_normals.py +690 -0
- diffusers-0.32.2/src/diffusers/pipelines/mochi/__init__.py +48 -0
- diffusers-0.32.2/src/diffusers/pipelines/mochi/pipeline_mochi.py +748 -0
- diffusers-0.32.2/src/diffusers/pipelines/mochi/pipeline_output.py +20 -0
- diffusers-0.32.2/src/diffusers/pipelines/musicldm/pipeline_musicldm.py +635 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/__init__.py +80 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pag_utils.py +243 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd.py +1328 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_inpaint.py +1543 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl.py +1610 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_xl_img2img.py +1683 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_hunyuandit.py +969 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_kolors.py +1136 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_pixart_sigma.py +865 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_sana.py +886 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_sd.py +1062 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_sd_3.py +994 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py +1058 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_sd_animatediff.py +866 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_sd_img2img.py +1094 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_sd_inpaint.py +1356 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_sd_xl.py +1345 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_img2img.py +1544 -0
- diffusers-0.32.2/src/diffusers/pipelines/pag/pipeline_pag_sd_xl_inpaint.py +1776 -0
- diffusers-0.32.2/src/diffusers/pipelines/paint_by_example/pipeline_paint_by_example.py +626 -0
- diffusers-0.32.2/src/diffusers/pipelines/pia/pipeline_pia.py +944 -0
- diffusers-0.32.2/src/diffusers/pipelines/pipeline_flax_utils.py +611 -0
- diffusers-0.32.2/src/diffusers/pipelines/pipeline_loading_utils.py +970 -0
- diffusers-0.32.2/src/diffusers/pipelines/pipeline_utils.py +1960 -0
- diffusers-0.32.2/src/diffusers/pipelines/pixart_alpha/__init__.py +55 -0
- diffusers-0.32.2/src/diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py +962 -0
- diffusers-0.32.2/src/diffusers/pipelines/pixart_alpha/pipeline_pixart_sigma.py +873 -0
- diffusers-0.32.2/src/diffusers/pipelines/sana/__init__.py +47 -0
- diffusers-0.32.2/src/diffusers/pipelines/sana/pipeline_output.py +21 -0
- diffusers-0.32.2/src/diffusers/pipelines/sana/pipeline_sana.py +884 -0
- diffusers-0.32.2/src/diffusers/pipelines/semantic_stable_diffusion/pipeline_semantic_stable_diffusion.py +722 -0
- diffusers-0.32.2/src/diffusers/pipelines/shap_e/pipeline_shap_e.py +334 -0
- diffusers-0.32.2/src/diffusers/pipelines/shap_e/pipeline_shap_e_img2img.py +321 -0
- diffusers-0.32.2/src/diffusers/pipelines/shap_e/renderer.py +1050 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_audio/__init__.py +50 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_audio/modeling_stable_audio.py +158 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_audio/pipeline_stable_audio.py +756 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade.py +528 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_combined.py +315 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_prior.py +639 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/__init__.py +202 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +1869 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion.py +473 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py +487 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_upscale.py +586 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py +1098 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py +875 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_image_variation.py +425 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py +1145 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py +1338 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py +906 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_latent_upscale.py +655 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py +809 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip.py +940 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py +846 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion/safety_checker.py +126 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_3/__init__.py +54 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_3/pipeline_output.py +21 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py +1140 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py +1036 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py +1250 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_attend_and_excite/pipeline_stable_diffusion_attend_and_excite.py +1097 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_diffedit/pipeline_stable_diffusion_diffedit.py +1531 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_gligen/pipeline_stable_diffusion_gligen.py +851 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_gligen/pipeline_stable_diffusion_gligen_text_image.py +1033 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_k_diffusion.py +670 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py +892 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py +1025 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py +1180 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_safe/pipeline_stable_diffusion_safe.py +769 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_safe/safety_checker.py +109 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_sag/pipeline_stable_diffusion_sag.py +954 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py +1304 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py +1498 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py +1736 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_instruct_pix2pix.py +992 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_diffusion_xl/watermark.py +42 -0
- diffusers-0.32.2/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py +726 -0
- diffusers-0.32.2/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_adapter.py +942 -0
- diffusers-0.32.2/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_xl_adapter.py +1289 -0
- diffusers-0.32.2/src/diffusers/pipelines/text_to_video_synthesis/pipeline_output.py +26 -0
- diffusers-0.32.2/src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth.py +643 -0
- diffusers-0.32.2/src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth_img2img.py +699 -0
- diffusers-0.32.2/src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero.py +981 -0
- diffusers-0.32.2/src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero_sdxl.py +1329 -0
- diffusers-0.32.2/src/diffusers/pipelines/unclip/pipeline_unclip.py +493 -0
- diffusers-0.32.2/src/diffusers/pipelines/unclip/pipeline_unclip_image_variation.py +420 -0
- diffusers-0.32.2/src/diffusers/pipelines/unidiffuser/modeling_text_decoder.py +296 -0
- diffusers-0.32.2/src/diffusers/pipelines/unidiffuser/modeling_uvit.py +1197 -0
- diffusers-0.32.2/src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py +1420 -0
- diffusers-0.32.2/src/diffusers/pipelines/wuerstchen/modeling_paella_vq_model.py +172 -0
- diffusers-0.32.2/src/diffusers/pipelines/wuerstchen/modeling_wuerstchen_common.py +76 -0
- diffusers-0.32.2/src/diffusers/pipelines/wuerstchen/modeling_wuerstchen_prior.py +198 -0
- diffusers-0.32.2/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen.py +438 -0
- diffusers-0.32.2/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen_combined.py +306 -0
- diffusers-0.32.2/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen_prior.py +517 -0
- diffusers-0.32.2/src/diffusers/quantizers/__init__.py +16 -0
- diffusers-0.32.2/src/diffusers/quantizers/auto.py +139 -0
- diffusers-0.32.2/src/diffusers/quantizers/base.py +233 -0
- diffusers-0.32.2/src/diffusers/quantizers/bitsandbytes/__init__.py +2 -0
- diffusers-0.32.2/src/diffusers/quantizers/bitsandbytes/bnb_quantizer.py +561 -0
- diffusers-0.32.2/src/diffusers/quantizers/bitsandbytes/utils.py +306 -0
- diffusers-0.32.2/src/diffusers/quantizers/gguf/__init__.py +1 -0
- diffusers-0.32.2/src/diffusers/quantizers/gguf/gguf_quantizer.py +159 -0
- diffusers-0.32.2/src/diffusers/quantizers/gguf/utils.py +456 -0
- diffusers-0.32.2/src/diffusers/quantizers/quantization_config.py +669 -0
- diffusers-0.32.2/src/diffusers/quantizers/torchao/__init__.py +15 -0
- diffusers-0.32.2/src/diffusers/quantizers/torchao/torchao_quantizer.py +292 -0
- diffusers-0.32.2/src/diffusers/schedulers/__init__.py +221 -0
- diffusers-0.32.2/src/diffusers/schedulers/deprecated/__init__.py +50 -0
- diffusers-0.32.2/src/diffusers/schedulers/deprecated/scheduling_karras_ve.py +243 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_amused.py +162 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_consistency_decoder.py +180 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_consistency_models.py +446 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_cosine_dpmsolver_multistep.py +572 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_ddim.py +521 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_ddim_cogvideox.py +452 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_ddim_flax.py +314 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_ddim_inverse.py +374 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_ddim_parallel.py +646 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_ddpm.py +559 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_ddpm_flax.py +303 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_ddpm_parallel.py +650 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_ddpm_wuerstchen.py +230 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_deis_multistep.py +886 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_dpm_cogvideox.py +489 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_dpmsolver_multistep.py +1166 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_dpmsolver_multistep_flax.py +643 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_dpmsolver_multistep_inverse.py +1027 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_dpmsolver_sde.py +676 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py +1168 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_edm_dpmsolver_multistep.py +707 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_edm_euler.py +404 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_euler_ancestral_discrete.py +482 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_euler_discrete.py +757 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py +458 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_flow_match_heun_discrete.py +320 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_heun_discrete.py +610 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_ipndm.py +224 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_k_dpm_2_ancestral_discrete.py +617 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_k_dpm_2_discrete.py +589 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_karras_ve_flax.py +238 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_lcm.py +654 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_lms_discrete.py +552 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_pndm.py +476 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_repaint.py +361 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_sasolver.py +1221 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_sde_ve.py +301 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_tcd.py +691 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_unclip.py +355 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_unipc_multistep.py +1076 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_utils.py +193 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_utils_flax.py +291 -0
- diffusers-0.32.2/src/diffusers/schedulers/scheduling_vq_diffusion.py +467 -0
- diffusers-0.32.2/src/diffusers/training_utils.py +637 -0
- diffusers-0.32.2/src/diffusers/utils/__init__.py +140 -0
- diffusers-0.32.2/src/diffusers/utils/constants.py +58 -0
- diffusers-0.32.2/src/diffusers/utils/doc_utils.py +39 -0
- diffusers-0.32.2/src/diffusers/utils/dummy_pt_objects.py +1755 -0
- diffusers-0.32.2/src/diffusers/utils/dummy_torch_and_torchsde_objects.py +32 -0
- diffusers-0.32.2/src/diffusers/utils/dummy_torch_and_transformers_and_sentencepiece_objects.py +47 -0
- diffusers-0.32.2/src/diffusers/utils/dummy_torch_and_transformers_objects.py +2537 -0
- diffusers-0.32.2/src/diffusers/utils/dynamic_modules_utils.py +457 -0
- diffusers-0.32.2/src/diffusers/utils/export_utils.py +184 -0
- diffusers-0.32.2/src/diffusers/utils/hub_utils.py +607 -0
- diffusers-0.32.2/src/diffusers/utils/import_utils.py +928 -0
- diffusers-0.32.2/src/diffusers/utils/loading_utils.py +162 -0
- diffusers-0.32.2/src/diffusers/utils/logging.py +341 -0
- diffusers-0.32.2/src/diffusers/utils/peft_utils.py +298 -0
- diffusers-0.32.2/src/diffusers/utils/state_dict_utils.py +335 -0
- diffusers-0.32.2/src/diffusers/utils/testing_utils.py +1149 -0
- diffusers-0.32.2/src/diffusers/utils/torch_utils.py +151 -0
- diffusers-0.32.2/src/diffusers/video_processor.py +113 -0
- diffusers-0.32.2/src/diffusers.egg-info/PKG-INFO +344 -0
- diffusers-0.32.2/src/diffusers.egg-info/SOURCES.txt +555 -0
- diffusers-0.32.2/src/diffusers.egg-info/requires.txt +84 -0
- diffusers-0.27.0/PKG-INFO +0 -279
- diffusers-0.27.0/README.md +0 -248
- diffusers-0.27.0/pyproject.toml +0 -27
- diffusers-0.27.0/setup.py +0 -281
- diffusers-0.27.0/src/diffusers/__init__.py +0 -787
- diffusers-0.27.0/src/diffusers/commands/env.py +0 -84
- diffusers-0.27.0/src/diffusers/configuration_utils.py +0 -703
- diffusers-0.27.0/src/diffusers/dependency_versions_table.py +0 -45
- diffusers-0.27.0/src/diffusers/image_processor.py +0 -990
- diffusers-0.27.0/src/diffusers/loaders/__init__.py +0 -88
- diffusers-0.27.0/src/diffusers/loaders/autoencoder.py +0 -146
- diffusers-0.27.0/src/diffusers/loaders/controlnet.py +0 -136
- diffusers-0.27.0/src/diffusers/loaders/ip_adapter.py +0 -281
- diffusers-0.27.0/src/diffusers/loaders/lora.py +0 -1349
- diffusers-0.27.0/src/diffusers/loaders/lora_conversion_utils.py +0 -284
- diffusers-0.27.0/src/diffusers/loaders/peft.py +0 -186
- diffusers-0.27.0/src/diffusers/loaders/single_file.py +0 -318
- diffusers-0.27.0/src/diffusers/loaders/single_file_utils.py +0 -1617
- diffusers-0.27.0/src/diffusers/loaders/textual_inversion.py +0 -562
- diffusers-0.27.0/src/diffusers/loaders/unet.py +0 -1003
- diffusers-0.27.0/src/diffusers/models/__init__.py +0 -103
- diffusers-0.27.0/src/diffusers/models/activations.py +0 -123
- diffusers-0.27.0/src/diffusers/models/adapter.py +0 -584
- diffusers-0.27.0/src/diffusers/models/attention.py +0 -665
- diffusers-0.27.0/src/diffusers/models/attention_flax.py +0 -494
- diffusers-0.27.0/src/diffusers/models/attention_processor.py +0 -2480
- diffusers-0.27.0/src/diffusers/models/autoencoders/__init__.py +0 -5
- diffusers-0.27.0/src/diffusers/models/autoencoders/autoencoder_asym_kl.py +0 -186
- diffusers-0.27.0/src/diffusers/models/autoencoders/autoencoder_kl.py +0 -489
- diffusers-0.27.0/src/diffusers/models/autoencoders/autoencoder_kl_temporal_decoder.py +0 -399
- diffusers-0.27.0/src/diffusers/models/autoencoders/autoencoder_tiny.py +0 -347
- diffusers-0.27.0/src/diffusers/models/autoencoders/consistency_decoder_vae.py +0 -435
- diffusers-0.27.0/src/diffusers/models/autoencoders/vae.py +0 -983
- diffusers-0.27.0/src/diffusers/models/controlnet.py +0 -868
- diffusers-0.27.0/src/diffusers/models/controlnet_flax.py +0 -395
- diffusers-0.27.0/src/diffusers/models/downsampling.py +0 -334
- diffusers-0.27.0/src/diffusers/models/dual_transformer_2d.py +0 -20
- diffusers-0.27.0/src/diffusers/models/embeddings.py +0 -914
- diffusers-0.27.0/src/diffusers/models/embeddings_flax.py +0 -97
- diffusers-0.27.0/src/diffusers/models/modeling_flax_pytorch_utils.py +0 -134
- diffusers-0.27.0/src/diffusers/models/modeling_flax_utils.py +0 -566
- diffusers-0.27.0/src/diffusers/models/modeling_outputs.py +0 -17
- diffusers-0.27.0/src/diffusers/models/modeling_pytorch_flax_utils.py +0 -161
- diffusers-0.27.0/src/diffusers/models/modeling_utils.py +0 -1021
- diffusers-0.27.0/src/diffusers/models/normalization.py +0 -254
- diffusers-0.27.0/src/diffusers/models/prior_transformer.py +0 -12
- diffusers-0.27.0/src/diffusers/models/resnet.py +0 -802
- diffusers-0.27.0/src/diffusers/models/t5_film_transformer.py +0 -70
- diffusers-0.27.0/src/diffusers/models/transformer_2d.py +0 -25
- diffusers-0.27.0/src/diffusers/models/transformer_temporal.py +0 -34
- diffusers-0.27.0/src/diffusers/models/transformers/__init__.py +0 -9
- diffusers-0.27.0/src/diffusers/models/transformers/dual_transformer_2d.py +0 -155
- diffusers-0.27.0/src/diffusers/models/transformers/prior_transformer.py +0 -380
- diffusers-0.27.0/src/diffusers/models/transformers/t5_film_transformer.py +0 -438
- diffusers-0.27.0/src/diffusers/models/transformers/transformer_2d.py +0 -456
- diffusers-0.27.0/src/diffusers/models/transformers/transformer_temporal.py +0 -379
- diffusers-0.27.0/src/diffusers/models/unet_1d.py +0 -26
- diffusers-0.27.0/src/diffusers/models/unet_1d_blocks.py +0 -203
- diffusers-0.27.0/src/diffusers/models/unet_2d.py +0 -27
- diffusers-0.27.0/src/diffusers/models/unet_2d_blocks.py +0 -375
- diffusers-0.27.0/src/diffusers/models/unet_2d_condition.py +0 -25
- diffusers-0.27.0/src/diffusers/models/unets/unet_1d.py +0 -255
- diffusers-0.27.0/src/diffusers/models/unets/unet_1d_blocks.py +0 -702
- diffusers-0.27.0/src/diffusers/models/unets/unet_2d.py +0 -346
- diffusers-0.27.0/src/diffusers/models/unets/unet_2d_blocks.py +0 -3731
- diffusers-0.27.0/src/diffusers/models/unets/unet_2d_condition.py +0 -1315
- diffusers-0.27.0/src/diffusers/models/unets/unet_2d_condition_flax.py +0 -453
- diffusers-0.27.0/src/diffusers/models/unets/unet_3d_blocks.py +0 -2405
- diffusers-0.27.0/src/diffusers/models/unets/unet_3d_condition.py +0 -753
- diffusers-0.27.0/src/diffusers/models/unets/unet_i2vgen_xl.py +0 -724
- diffusers-0.27.0/src/diffusers/models/unets/unet_kandinsky3.py +0 -535
- diffusers-0.27.0/src/diffusers/models/unets/unet_motion_model.py +0 -948
- diffusers-0.27.0/src/diffusers/models/unets/unet_spatio_temporal_condition.py +0 -489
- diffusers-0.27.0/src/diffusers/models/unets/unet_stable_cascade.py +0 -610
- diffusers-0.27.0/src/diffusers/models/unets/uvit_2d.py +0 -470
- diffusers-0.27.0/src/diffusers/models/upsampling.py +0 -448
- diffusers-0.27.0/src/diffusers/models/vq_model.py +0 -181
- diffusers-0.27.0/src/diffusers/optimization.py +0 -361
- diffusers-0.27.0/src/diffusers/pipelines/__init__.py +0 -581
- diffusers-0.27.0/src/diffusers/pipelines/amused/pipeline_amused.py +0 -328
- diffusers-0.27.0/src/diffusers/pipelines/amused/pipeline_amused_img2img.py +0 -347
- diffusers-0.27.0/src/diffusers/pipelines/amused/pipeline_amused_inpaint.py +0 -378
- diffusers-0.27.0/src/diffusers/pipelines/animatediff/__init__.py +0 -49
- diffusers-0.27.0/src/diffusers/pipelines/animatediff/pipeline_animatediff.py +0 -847
- diffusers-0.27.0/src/diffusers/pipelines/animatediff/pipeline_animatediff_video2video.py +0 -997
- diffusers-0.27.0/src/diffusers/pipelines/animatediff/pipeline_output.py +0 -23
- diffusers-0.27.0/src/diffusers/pipelines/audioldm/pipeline_audioldm.py +0 -546
- diffusers-0.27.0/src/diffusers/pipelines/audioldm2/modeling_audioldm2.py +0 -1511
- diffusers-0.27.0/src/diffusers/pipelines/audioldm2/pipeline_audioldm2.py +0 -980
- diffusers-0.27.0/src/diffusers/pipelines/auto_pipeline.py +0 -987
- diffusers-0.27.0/src/diffusers/pipelines/blip_diffusion/blip_image_processing.py +0 -318
- diffusers-0.27.0/src/diffusers/pipelines/blip_diffusion/modeling_blip2.py +0 -642
- diffusers-0.27.0/src/diffusers/pipelines/blip_diffusion/modeling_ctx_clip.py +0 -223
- diffusers-0.27.0/src/diffusers/pipelines/blip_diffusion/pipeline_blip_diffusion.py +0 -348
- diffusers-0.27.0/src/diffusers/pipelines/consistency_models/pipeline_consistency_models.py +0 -275
- diffusers-0.27.0/src/diffusers/pipelines/controlnet/__init__.py +0 -80
- diffusers-0.27.0/src/diffusers/pipelines/controlnet/multicontrolnet.py +0 -187
- diffusers-0.27.0/src/diffusers/pipelines/controlnet/pipeline_controlnet.py +0 -1318
- diffusers-0.27.0/src/diffusers/pipelines/controlnet/pipeline_controlnet_blip_diffusion.py +0 -413
- diffusers-0.27.0/src/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py +0 -1310
- diffusers-0.27.0/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py +0 -1620
- diffusers-0.27.0/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py +0 -1818
- diffusers-0.27.0/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py +0 -1499
- diffusers-0.27.0/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py +0 -1626
- diffusers-0.27.0/src/diffusers/pipelines/controlnet/pipeline_flax_controlnet.py +0 -532
- diffusers-0.27.0/src/diffusers/pipelines/ddpm/pipeline_ddpm.py +0 -127
- diffusers-0.27.0/src/diffusers/pipelines/deepfloyd_if/pipeline_if.py +0 -788
- diffusers-0.27.0/src/diffusers/pipelines/deepfloyd_if/pipeline_if_img2img.py +0 -910
- diffusers-0.27.0/src/diffusers/pipelines/deepfloyd_if/pipeline_if_img2img_superresolution.py +0 -1029
- diffusers-0.27.0/src/diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting.py +0 -1030
- diffusers-0.27.0/src/diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting_superresolution.py +0 -1137
- diffusers-0.27.0/src/diffusers/pipelines/deepfloyd_if/pipeline_if_superresolution.py +0 -885
- diffusers-0.27.0/src/diffusers/pipelines/deepfloyd_if/pipeline_output.py +0 -28
- diffusers-0.27.0/src/diffusers/pipelines/deepfloyd_if/watermark.py +0 -46
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/alt_diffusion/modeling_roberta_series.py +0 -124
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion.py +0 -946
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_alt_diffusion_img2img.py +0 -1018
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/repaint/pipeline_repaint.py +0 -230
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/spectrogram_diffusion/pipeline_spectrogram_diffusion.py +0 -269
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_cycle_diffusion.py +0 -948
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_inpaint_legacy.py +0 -786
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_model_editing.py +0 -824
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_paradigms.py +0 -786
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_pix2pix_zero.py +0 -1304
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/versatile_diffusion/modeling_text_unet.py +0 -2508
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion.py +0 -421
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion_dual_guided.py +0 -556
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion_image_variation.py +0 -397
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/versatile_diffusion/pipeline_versatile_diffusion_text_to_image.py +0 -475
- diffusers-0.27.0/src/diffusers/pipelines/deprecated/vq_diffusion/pipeline_vq_diffusion.py +0 -325
- diffusers-0.27.0/src/diffusers/pipelines/dit/pipeline_dit.py +0 -233
- diffusers-0.27.0/src/diffusers/pipelines/free_init_utils.py +0 -184
- diffusers-0.27.0/src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py +0 -798
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky/pipeline_kandinsky.py +0 -407
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky/pipeline_kandinsky_combined.py +0 -814
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky/pipeline_kandinsky_img2img.py +0 -500
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky/pipeline_kandinsky_inpaint.py +0 -635
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky/pipeline_kandinsky_prior.py +0 -547
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2.py +0 -320
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_combined.py +0 -851
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_controlnet.py +0 -320
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_controlnet_img2img.py +0 -381
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_img2img.py +0 -399
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_inpainting.py +0 -556
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior.py +0 -549
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior_emb2emb.py +0 -563
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky3/convert_kandinsky3_unet.py +0 -98
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3.py +0 -589
- diffusers-0.27.0/src/diffusers/pipelines/kandinsky3/pipeline_kandinsky3_img2img.py +0 -654
- diffusers-0.27.0/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_img2img.py +0 -956
- diffusers-0.27.0/src/diffusers/pipelines/latent_consistency_models/pipeline_latent_consistency_text2img.py +0 -888
- diffusers-0.27.0/src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py +0 -746
- diffusers-0.27.0/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py +0 -1505
- diffusers-0.27.0/src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py +0 -1797
- diffusers-0.27.0/src/diffusers/pipelines/ledits_pp/pipeline_output.py +0 -43
- diffusers-0.27.0/src/diffusers/pipelines/musicldm/pipeline_musicldm.py +0 -635
- diffusers-0.27.0/src/diffusers/pipelines/paint_by_example/pipeline_paint_by_example.py +0 -621
- diffusers-0.27.0/src/diffusers/pipelines/pia/pipeline_pia.py +0 -1034
- diffusers-0.27.0/src/diffusers/pipelines/pipeline_flax_utils.py +0 -616
- diffusers-0.27.0/src/diffusers/pipelines/pipeline_loading_utils.py +0 -508
- diffusers-0.27.0/src/diffusers/pipelines/pipeline_utils.py +0 -1771
- diffusers-0.27.0/src/diffusers/pipelines/pixart_alpha/__init__.py +0 -48
- diffusers-0.27.0/src/diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py +0 -979
- diffusers-0.27.0/src/diffusers/pipelines/semantic_stable_diffusion/pipeline_semantic_stable_diffusion.py +0 -718
- diffusers-0.27.0/src/diffusers/pipelines/shap_e/pipeline_shap_e.py +0 -334
- diffusers-0.27.0/src/diffusers/pipelines/shap_e/pipeline_shap_e_img2img.py +0 -321
- diffusers-0.27.0/src/diffusers/pipelines/shap_e/renderer.py +0 -1050
- diffusers-0.27.0/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade.py +0 -482
- diffusers-0.27.0/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_combined.py +0 -311
- diffusers-0.27.0/src/diffusers/pipelines/stable_cascade/pipeline_stable_cascade_prior.py +0 -638
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/__init__.py +0 -203
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +0 -1860
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion.py +0 -473
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py +0 -487
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_upscale.py +0 -586
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py +0 -1032
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py +0 -860
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_image_variation.py +0 -420
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py +0 -1113
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py +0 -1430
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py +0 -807
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_latent_upscale.py +0 -495
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py +0 -808
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip.py +0 -932
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py +0 -839
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion/safety_checker.py +0 -125
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_attend_and_excite/pipeline_stable_diffusion_attend_and_excite.py +0 -1088
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_diffedit/pipeline_stable_diffusion_diffedit.py +0 -1530
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_gligen/pipeline_stable_diffusion_gligen.py +0 -845
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_gligen/pipeline_stable_diffusion_gligen_text_image.py +0 -1017
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_k_diffusion.py +0 -664
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py +0 -891
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py +0 -985
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_panorama/pipeline_stable_diffusion_panorama.py +0 -933
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_safe/pipeline_stable_diffusion_safe.py +0 -764
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_safe/safety_checker.py +0 -109
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_sag/pipeline_stable_diffusion_sag.py +0 -886
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py +0 -1266
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py +0 -1442
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py +0 -1812
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_instruct_pix2pix.py +0 -976
- diffusers-0.27.0/src/diffusers/pipelines/stable_diffusion_xl/watermark.py +0 -36
- diffusers-0.27.0/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py +0 -673
- diffusers-0.27.0/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_adapter.py +0 -912
- diffusers-0.27.0/src/diffusers/pipelines/t2i_adapter/pipeline_stable_diffusion_xl_adapter.py +0 -1258
- diffusers-0.27.0/src/diffusers/pipelines/text_to_video_synthesis/pipeline_output.py +0 -25
- diffusers-0.27.0/src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth.py +0 -663
- diffusers-0.27.0/src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth_img2img.py +0 -760
- diffusers-0.27.0/src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero.py +0 -969
- diffusers-0.27.0/src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero_sdxl.py +0 -1315
- diffusers-0.27.0/src/diffusers/pipelines/unclip/pipeline_unclip.py +0 -493
- diffusers-0.27.0/src/diffusers/pipelines/unclip/pipeline_unclip_image_variation.py +0 -420
- diffusers-0.27.0/src/diffusers/pipelines/unidiffuser/modeling_text_decoder.py +0 -296
- diffusers-0.27.0/src/diffusers/pipelines/unidiffuser/modeling_uvit.py +0 -1197
- diffusers-0.27.0/src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py +0 -1419
- diffusers-0.27.0/src/diffusers/pipelines/wuerstchen/modeling_paella_vq_model.py +0 -172
- diffusers-0.27.0/src/diffusers/pipelines/wuerstchen/modeling_wuerstchen_common.py +0 -81
- diffusers-0.27.0/src/diffusers/pipelines/wuerstchen/modeling_wuerstchen_prior.py +0 -200
- diffusers-0.27.0/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen.py +0 -438
- diffusers-0.27.0/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen_combined.py +0 -306
- diffusers-0.27.0/src/diffusers/pipelines/wuerstchen/pipeline_wuerstchen_prior.py +0 -516
- diffusers-0.27.0/src/diffusers/schedulers/__init__.py +0 -211
- diffusers-0.27.0/src/diffusers/schedulers/deprecated/__init__.py +0 -50
- diffusers-0.27.0/src/diffusers/schedulers/deprecated/scheduling_karras_ve.py +0 -243
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_amused.py +0 -162
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_consistency_decoder.py +0 -180
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_consistency_models.py +0 -448
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_ddim.py +0 -520
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_ddim_flax.py +0 -313
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_ddim_inverse.py +0 -374
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_ddim_parallel.py +0 -645
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_ddpm.py +0 -562
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_ddpm_flax.py +0 -299
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_ddpm_parallel.py +0 -653
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_ddpm_wuerstchen.py +0 -230
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_deis_multistep.py +0 -786
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_dpmsolver_multistep.py +0 -1029
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_dpmsolver_multistep_flax.py +0 -643
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_dpmsolver_multistep_inverse.py +0 -921
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_dpmsolver_sde.py +0 -557
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py +0 -979
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_edm_dpmsolver_multistep.py +0 -683
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_edm_euler.py +0 -381
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_euler_ancestral_discrete.py +0 -481
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_euler_discrete.py +0 -576
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_heun_discrete.py +0 -482
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_ipndm.py +0 -224
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_k_dpm_2_ancestral_discrete.py +0 -508
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_k_dpm_2_discrete.py +0 -483
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_karras_ve_flax.py +0 -238
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_lcm.py +0 -660
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_lms_discrete.py +0 -475
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_pndm.py +0 -476
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_repaint.py +0 -361
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_sasolver.py +0 -1124
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_sde_ve.py +0 -301
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_tcd.py +0 -686
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_unclip.py +0 -352
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_unipc_multistep.py +0 -880
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_utils.py +0 -186
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_utils_flax.py +0 -293
- diffusers-0.27.0/src/diffusers/schedulers/scheduling_vq_diffusion.py +0 -467
- diffusers-0.27.0/src/diffusers/training_utils.py +0 -453
- diffusers-0.27.0/src/diffusers/utils/__init__.py +0 -124
- diffusers-0.27.0/src/diffusers/utils/constants.py +0 -55
- diffusers-0.27.0/src/diffusers/utils/doc_utils.py +0 -38
- diffusers-0.27.0/src/diffusers/utils/dummy_pt_objects.py +0 -1170
- diffusers-0.27.0/src/diffusers/utils/dummy_torch_and_torchsde_objects.py +0 -17
- diffusers-0.27.0/src/diffusers/utils/dummy_torch_and_transformers_objects.py +0 -1607
- diffusers-0.27.0/src/diffusers/utils/dynamic_modules_utils.py +0 -452
- diffusers-0.27.0/src/diffusers/utils/export_utils.py +0 -140
- diffusers-0.27.0/src/diffusers/utils/hub_utils.py +0 -493
- diffusers-0.27.0/src/diffusers/utils/import_utils.py +0 -726
- diffusers-0.27.0/src/diffusers/utils/loading_utils.py +0 -49
- diffusers-0.27.0/src/diffusers/utils/logging.py +0 -339
- diffusers-0.27.0/src/diffusers/utils/peft_utils.py +0 -268
- diffusers-0.27.0/src/diffusers/utils/state_dict_utils.py +0 -324
- diffusers-0.27.0/src/diffusers/utils/testing_utils.py +0 -967
- diffusers-0.27.0/src/diffusers/utils/torch_utils.py +0 -147
- diffusers-0.27.0/src/diffusers.egg-info/PKG-INFO +0 -279
- diffusers-0.27.0/src/diffusers.egg-info/SOURCES.txt +0 -404
- diffusers-0.27.0/src/diffusers.egg-info/requires.txt +0 -84
- {diffusers-0.27.0 → diffusers-0.32.2}/LICENSE +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/MANIFEST.in +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/setup.cfg +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/commands/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/commands/diffusers_cli.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/commands/fp16_safetensors.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/dependency_versions_check.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/experimental/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/experimental/rl/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/experimental/rl/value_guided_sampling.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/loaders/utils.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/models/lora.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/models/resnet_flax.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/models/unets/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/models/unets/unet_2d_blocks_flax.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/models/vae_flax.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/amused/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/audioldm/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/audioldm2/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/blip_diffusion/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/consistency_models/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/dance_diffusion/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/dance_diffusion/pipeline_dance_diffusion.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/ddim/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/ddim/pipeline_ddim.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/ddpm/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deepfloyd_if/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deepfloyd_if/safety_checker.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deepfloyd_if/timesteps.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/alt_diffusion/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/alt_diffusion/pipeline_output.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/audio_diffusion/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/audio_diffusion/mel.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/audio_diffusion/pipeline_audio_diffusion.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/latent_diffusion_uncond/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/latent_diffusion_uncond/pipeline_latent_diffusion_uncond.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/pndm/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/pndm/pipeline_pndm.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/repaint/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/score_sde_ve/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/score_sde_ve/pipeline_score_sde_ve.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/spectrogram_diffusion/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/spectrogram_diffusion/continuous_encoder.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/spectrogram_diffusion/midi_utils.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/spectrogram_diffusion/notes_encoder.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/stable_diffusion_variants/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_onnx_stable_diffusion_inpaint_legacy.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/stochastic_karras_ve/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/stochastic_karras_ve/pipeline_stochastic_karras_ve.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/versatile_diffusion/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/deprecated/vq_diffusion/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/dit/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/i2vgen_xl/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/kandinsky/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/kandinsky/text_encoder.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/kandinsky2_2/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/kandinsky3/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/latent_consistency_models/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/latent_diffusion/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion_superresolution.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/ledits_pp/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/musicldm/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/onnx_utils.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/paint_by_example/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/paint_by_example/image_encoder.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/pia/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/semantic_stable_diffusion/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/semantic_stable_diffusion/pipeline_output.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/shap_e/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/shap_e/camera.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_cascade/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion/clip_image_project_model.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion_img2img.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion_inpaint.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_img2img.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion/pipeline_output.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion/safety_checker_flax.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion/stable_unclip_image_normalizer.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion_attend_and_excite/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion_diffedit/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion_gligen/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion_k_diffusion/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion_ldm3d/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion_panorama/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion_safe/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion_safe/pipeline_output.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion_sag/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion_xl/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion_xl/pipeline_flax_stable_diffusion_xl.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_diffusion_xl/pipeline_output.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/stable_video_diffusion/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/t2i_adapter/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/text_to_video_synthesis/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/unclip/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/unclip/text_proj.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/unidiffuser/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/wuerstchen/__init__.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/pipelines/wuerstchen/modeling_wuerstchen_diffnext.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/py.typed +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/schedulers/deprecated/scheduling_sde_vp.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/schedulers/scheduling_euler_discrete_flax.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/schedulers/scheduling_lms_discrete_flax.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/schedulers/scheduling_pndm_flax.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/schedulers/scheduling_sde_ve_flax.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/utils/accelerate_utils.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/utils/deprecation_utils.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/utils/dummy_flax_and_transformers_objects.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/utils/dummy_flax_objects.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/utils/dummy_note_seq_objects.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/utils/dummy_onnx_objects.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/utils/dummy_torch_and_librosa_objects.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/utils/dummy_torch_and_scipy_objects.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/utils/dummy_torch_and_transformers_and_k_diffusion_objects.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/utils/dummy_torch_and_transformers_and_onnx_objects.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/utils/dummy_transformers_and_torch_and_note_seq_objects.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/utils/model_card_template.md +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/utils/outputs.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/utils/pil_utils.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers/utils/versions.py +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers.egg-info/dependency_links.txt +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers.egg-info/entry_points.txt +0 -0
- {diffusers-0.27.0 → diffusers-0.32.2}/src/diffusers.egg-info/top_level.txt +0 -0
@@ -0,0 +1,344 @@
|
|
1
|
+
Metadata-Version: 2.1
|
2
|
+
Name: diffusers
|
3
|
+
Version: 0.32.2
|
4
|
+
Summary: State-of-the-art diffusion in PyTorch and JAX.
|
5
|
+
Home-page: https://github.com/huggingface/diffusers
|
6
|
+
Author: The Hugging Face team (past and future) with the help of all our contributors (https://github.com/huggingface/diffusers/graphs/contributors)
|
7
|
+
Author-email: diffusers@huggingface.co
|
8
|
+
License: Apache 2.0 License
|
9
|
+
Keywords: deep learning diffusion jax pytorch stable diffusion audioldm
|
10
|
+
Classifier: Development Status :: 5 - Production/Stable
|
11
|
+
Classifier: Intended Audience :: Developers
|
12
|
+
Classifier: Intended Audience :: Education
|
13
|
+
Classifier: Intended Audience :: Science/Research
|
14
|
+
Classifier: License :: OSI Approved :: Apache Software License
|
15
|
+
Classifier: Operating System :: OS Independent
|
16
|
+
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
|
17
|
+
Classifier: Programming Language :: Python :: 3
|
18
|
+
Classifier: Programming Language :: Python :: 3.8
|
19
|
+
Classifier: Programming Language :: Python :: 3.9
|
20
|
+
Classifier: Programming Language :: Python :: 3.10
|
21
|
+
Requires-Python: >=3.8.0
|
22
|
+
Description-Content-Type: text/markdown
|
23
|
+
License-File: LICENSE
|
24
|
+
Requires-Dist: importlib_metadata
|
25
|
+
Requires-Dist: filelock
|
26
|
+
Requires-Dist: huggingface-hub>=0.23.2
|
27
|
+
Requires-Dist: numpy
|
28
|
+
Requires-Dist: regex!=2019.12.17
|
29
|
+
Requires-Dist: requests
|
30
|
+
Requires-Dist: safetensors>=0.3.1
|
31
|
+
Requires-Dist: Pillow
|
32
|
+
Provides-Extra: quality
|
33
|
+
Requires-Dist: urllib3<=2.0.0; extra == "quality"
|
34
|
+
Requires-Dist: isort>=5.5.4; extra == "quality"
|
35
|
+
Requires-Dist: ruff==0.1.5; extra == "quality"
|
36
|
+
Requires-Dist: hf-doc-builder>=0.3.0; extra == "quality"
|
37
|
+
Provides-Extra: docs
|
38
|
+
Requires-Dist: hf-doc-builder>=0.3.0; extra == "docs"
|
39
|
+
Provides-Extra: training
|
40
|
+
Requires-Dist: accelerate>=0.31.0; extra == "training"
|
41
|
+
Requires-Dist: datasets; extra == "training"
|
42
|
+
Requires-Dist: protobuf<4,>=3.20.3; extra == "training"
|
43
|
+
Requires-Dist: tensorboard; extra == "training"
|
44
|
+
Requires-Dist: Jinja2; extra == "training"
|
45
|
+
Requires-Dist: peft>=0.6.0; extra == "training"
|
46
|
+
Provides-Extra: test
|
47
|
+
Requires-Dist: compel==0.1.8; extra == "test"
|
48
|
+
Requires-Dist: GitPython<3.1.19; extra == "test"
|
49
|
+
Requires-Dist: datasets; extra == "test"
|
50
|
+
Requires-Dist: Jinja2; extra == "test"
|
51
|
+
Requires-Dist: invisible-watermark>=0.2.0; extra == "test"
|
52
|
+
Requires-Dist: k-diffusion>=0.0.12; extra == "test"
|
53
|
+
Requires-Dist: librosa; extra == "test"
|
54
|
+
Requires-Dist: parameterized; extra == "test"
|
55
|
+
Requires-Dist: pytest; extra == "test"
|
56
|
+
Requires-Dist: pytest-timeout; extra == "test"
|
57
|
+
Requires-Dist: pytest-xdist; extra == "test"
|
58
|
+
Requires-Dist: requests-mock==1.10.0; extra == "test"
|
59
|
+
Requires-Dist: safetensors>=0.3.1; extra == "test"
|
60
|
+
Requires-Dist: sentencepiece!=0.1.92,>=0.1.91; extra == "test"
|
61
|
+
Requires-Dist: scipy; extra == "test"
|
62
|
+
Requires-Dist: torchvision; extra == "test"
|
63
|
+
Requires-Dist: transformers>=4.41.2; extra == "test"
|
64
|
+
Provides-Extra: torch
|
65
|
+
Requires-Dist: torch>=1.4; extra == "torch"
|
66
|
+
Requires-Dist: accelerate>=0.31.0; extra == "torch"
|
67
|
+
Provides-Extra: flax
|
68
|
+
Requires-Dist: jax>=0.4.1; extra == "flax"
|
69
|
+
Requires-Dist: jaxlib>=0.4.1; extra == "flax"
|
70
|
+
Requires-Dist: flax>=0.4.1; extra == "flax"
|
71
|
+
Provides-Extra: dev
|
72
|
+
Requires-Dist: urllib3<=2.0.0; extra == "dev"
|
73
|
+
Requires-Dist: isort>=5.5.4; extra == "dev"
|
74
|
+
Requires-Dist: ruff==0.1.5; extra == "dev"
|
75
|
+
Requires-Dist: hf-doc-builder>=0.3.0; extra == "dev"
|
76
|
+
Requires-Dist: compel==0.1.8; extra == "dev"
|
77
|
+
Requires-Dist: GitPython<3.1.19; extra == "dev"
|
78
|
+
Requires-Dist: datasets; extra == "dev"
|
79
|
+
Requires-Dist: Jinja2; extra == "dev"
|
80
|
+
Requires-Dist: invisible-watermark>=0.2.0; extra == "dev"
|
81
|
+
Requires-Dist: k-diffusion>=0.0.12; extra == "dev"
|
82
|
+
Requires-Dist: librosa; extra == "dev"
|
83
|
+
Requires-Dist: parameterized; extra == "dev"
|
84
|
+
Requires-Dist: pytest; extra == "dev"
|
85
|
+
Requires-Dist: pytest-timeout; extra == "dev"
|
86
|
+
Requires-Dist: pytest-xdist; extra == "dev"
|
87
|
+
Requires-Dist: requests-mock==1.10.0; extra == "dev"
|
88
|
+
Requires-Dist: safetensors>=0.3.1; extra == "dev"
|
89
|
+
Requires-Dist: sentencepiece!=0.1.92,>=0.1.91; extra == "dev"
|
90
|
+
Requires-Dist: scipy; extra == "dev"
|
91
|
+
Requires-Dist: torchvision; extra == "dev"
|
92
|
+
Requires-Dist: transformers>=4.41.2; extra == "dev"
|
93
|
+
Requires-Dist: accelerate>=0.31.0; extra == "dev"
|
94
|
+
Requires-Dist: datasets; extra == "dev"
|
95
|
+
Requires-Dist: protobuf<4,>=3.20.3; extra == "dev"
|
96
|
+
Requires-Dist: tensorboard; extra == "dev"
|
97
|
+
Requires-Dist: Jinja2; extra == "dev"
|
98
|
+
Requires-Dist: peft>=0.6.0; extra == "dev"
|
99
|
+
Requires-Dist: hf-doc-builder>=0.3.0; extra == "dev"
|
100
|
+
Requires-Dist: torch>=1.4; extra == "dev"
|
101
|
+
Requires-Dist: accelerate>=0.31.0; extra == "dev"
|
102
|
+
Requires-Dist: jax>=0.4.1; extra == "dev"
|
103
|
+
Requires-Dist: jaxlib>=0.4.1; extra == "dev"
|
104
|
+
Requires-Dist: flax>=0.4.1; extra == "dev"
|
105
|
+
|
106
|
+
<!---
|
107
|
+
Copyright 2022 - The HuggingFace Team. All rights reserved.
|
108
|
+
|
109
|
+
Licensed under the Apache License, Version 2.0 (the "License");
|
110
|
+
you may not use this file except in compliance with the License.
|
111
|
+
You may obtain a copy of the License at
|
112
|
+
|
113
|
+
http://www.apache.org/licenses/LICENSE-2.0
|
114
|
+
|
115
|
+
Unless required by applicable law or agreed to in writing, software
|
116
|
+
distributed under the License is distributed on an "AS IS" BASIS,
|
117
|
+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
118
|
+
See the License for the specific language governing permissions and
|
119
|
+
limitations under the License.
|
120
|
+
-->
|
121
|
+
|
122
|
+
<p align="center">
|
123
|
+
<br>
|
124
|
+
<img src="https://raw.githubusercontent.com/huggingface/diffusers/main/docs/source/en/imgs/diffusers_library.jpg" width="400"/>
|
125
|
+
<br>
|
126
|
+
<p>
|
127
|
+
<p align="center">
|
128
|
+
<a href="https://github.com/huggingface/diffusers/blob/main/LICENSE"><img alt="GitHub" src="https://img.shields.io/github/license/huggingface/datasets.svg?color=blue"></a>
|
129
|
+
<a href="https://github.com/huggingface/diffusers/releases"><img alt="GitHub release" src="https://img.shields.io/github/release/huggingface/diffusers.svg"></a>
|
130
|
+
<a href="https://pepy.tech/project/diffusers"><img alt="GitHub release" src="https://static.pepy.tech/badge/diffusers/month"></a>
|
131
|
+
<a href="CODE_OF_CONDUCT.md"><img alt="Contributor Covenant" src="https://img.shields.io/badge/Contributor%20Covenant-2.1-4baaaa.svg"></a>
|
132
|
+
<a href="https://twitter.com/diffuserslib"><img alt="X account" src="https://img.shields.io/twitter/url/https/twitter.com/diffuserslib.svg?style=social&label=Follow%20%40diffuserslib"></a>
|
133
|
+
</p>
|
134
|
+
|
135
|
+
🤗 Diffusers is the go-to library for state-of-the-art pretrained diffusion models for generating images, audio, and even 3D structures of molecules. Whether you're looking for a simple inference solution or training your own diffusion models, 🤗 Diffusers is a modular toolbox that supports both. Our library is designed with a focus on [usability over performance](https://huggingface.co/docs/diffusers/conceptual/philosophy#usability-over-performance), [simple over easy](https://huggingface.co/docs/diffusers/conceptual/philosophy#simple-over-easy), and [customizability over abstractions](https://huggingface.co/docs/diffusers/conceptual/philosophy#tweakable-contributorfriendly-over-abstraction).
|
136
|
+
|
137
|
+
🤗 Diffusers offers three core components:
|
138
|
+
|
139
|
+
- State-of-the-art [diffusion pipelines](https://huggingface.co/docs/diffusers/api/pipelines/overview) that can be run in inference with just a few lines of code.
|
140
|
+
- Interchangeable noise [schedulers](https://huggingface.co/docs/diffusers/api/schedulers/overview) for different diffusion speeds and output quality.
|
141
|
+
- Pretrained [models](https://huggingface.co/docs/diffusers/api/models/overview) that can be used as building blocks, and combined with schedulers, for creating your own end-to-end diffusion systems.
|
142
|
+
|
143
|
+
## Installation
|
144
|
+
|
145
|
+
We recommend installing 🤗 Diffusers in a virtual environment from PyPI or Conda. For more details about installing [PyTorch](https://pytorch.org/get-started/locally/) and [Flax](https://flax.readthedocs.io/en/latest/#installation), please refer to their official documentation.
|
146
|
+
|
147
|
+
### PyTorch
|
148
|
+
|
149
|
+
With `pip` (official package):
|
150
|
+
|
151
|
+
```bash
|
152
|
+
pip install --upgrade diffusers[torch]
|
153
|
+
```
|
154
|
+
|
155
|
+
With `conda` (maintained by the community):
|
156
|
+
|
157
|
+
```sh
|
158
|
+
conda install -c conda-forge diffusers
|
159
|
+
```
|
160
|
+
|
161
|
+
### Flax
|
162
|
+
|
163
|
+
With `pip` (official package):
|
164
|
+
|
165
|
+
```bash
|
166
|
+
pip install --upgrade diffusers[flax]
|
167
|
+
```
|
168
|
+
|
169
|
+
### Apple Silicon (M1/M2) support
|
170
|
+
|
171
|
+
Please refer to the [How to use Stable Diffusion in Apple Silicon](https://huggingface.co/docs/diffusers/optimization/mps) guide.
|
172
|
+
|
173
|
+
## Quickstart
|
174
|
+
|
175
|
+
Generating outputs is super easy with 🤗 Diffusers. To generate an image from text, use the `from_pretrained` method to load any pretrained diffusion model (browse the [Hub](https://huggingface.co/models?library=diffusers&sort=downloads) for 30,000+ checkpoints):
|
176
|
+
|
177
|
+
```python
|
178
|
+
from diffusers import DiffusionPipeline
|
179
|
+
import torch
|
180
|
+
|
181
|
+
pipeline = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch.float16)
|
182
|
+
pipeline.to("cuda")
|
183
|
+
pipeline("An image of a squirrel in Picasso style").images[0]
|
184
|
+
```
|
185
|
+
|
186
|
+
You can also dig into the models and schedulers toolbox to build your own diffusion system:
|
187
|
+
|
188
|
+
```python
|
189
|
+
from diffusers import DDPMScheduler, UNet2DModel
|
190
|
+
from PIL import Image
|
191
|
+
import torch
|
192
|
+
|
193
|
+
scheduler = DDPMScheduler.from_pretrained("google/ddpm-cat-256")
|
194
|
+
model = UNet2DModel.from_pretrained("google/ddpm-cat-256").to("cuda")
|
195
|
+
scheduler.set_timesteps(50)
|
196
|
+
|
197
|
+
sample_size = model.config.sample_size
|
198
|
+
noise = torch.randn((1, 3, sample_size, sample_size), device="cuda")
|
199
|
+
input = noise
|
200
|
+
|
201
|
+
for t in scheduler.timesteps:
|
202
|
+
with torch.no_grad():
|
203
|
+
noisy_residual = model(input, t).sample
|
204
|
+
prev_noisy_sample = scheduler.step(noisy_residual, t, input).prev_sample
|
205
|
+
input = prev_noisy_sample
|
206
|
+
|
207
|
+
image = (input / 2 + 0.5).clamp(0, 1)
|
208
|
+
image = image.cpu().permute(0, 2, 3, 1).numpy()[0]
|
209
|
+
image = Image.fromarray((image * 255).round().astype("uint8"))
|
210
|
+
image
|
211
|
+
```
|
212
|
+
|
213
|
+
Check out the [Quickstart](https://huggingface.co/docs/diffusers/quicktour) to launch your diffusion journey today!
|
214
|
+
|
215
|
+
## How to navigate the documentation
|
216
|
+
|
217
|
+
| **Documentation** | **What can I learn?** |
|
218
|
+
|---------------------------------------------------------------------|-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
219
|
+
| [Tutorial](https://huggingface.co/docs/diffusers/tutorials/tutorial_overview) | A basic crash course for learning how to use the library's most important features like using models and schedulers to build your own diffusion system, and training your own diffusion model. |
|
220
|
+
| [Loading](https://huggingface.co/docs/diffusers/using-diffusers/loading) | Guides for how to load and configure all the components (pipelines, models, and schedulers) of the library, as well as how to use different schedulers. |
|
221
|
+
| [Pipelines for inference](https://huggingface.co/docs/diffusers/using-diffusers/overview_techniques) | Guides for how to use pipelines for different inference tasks, batched generation, controlling generated outputs and randomness, and how to contribute a pipeline to the library. |
|
222
|
+
| [Optimization](https://huggingface.co/docs/diffusers/optimization/fp16) | Guides for how to optimize your diffusion model to run faster and consume less memory. |
|
223
|
+
| [Training](https://huggingface.co/docs/diffusers/training/overview) | Guides for how to train a diffusion model for different tasks with different training techniques. |
|
224
|
+
## Contribution
|
225
|
+
|
226
|
+
We ❤️ contributions from the open-source community!
|
227
|
+
If you want to contribute to this library, please check out our [Contribution guide](https://github.com/huggingface/diffusers/blob/main/CONTRIBUTING.md).
|
228
|
+
You can look out for [issues](https://github.com/huggingface/diffusers/issues) you'd like to tackle to contribute to the library.
|
229
|
+
- See [Good first issues](https://github.com/huggingface/diffusers/issues?q=is%3Aopen+is%3Aissue+label%3A%22good+first+issue%22) for general opportunities to contribute
|
230
|
+
- See [New model/pipeline](https://github.com/huggingface/diffusers/issues?q=is%3Aopen+is%3Aissue+label%3A%22New+pipeline%2Fmodel%22) to contribute exciting new diffusion models / diffusion pipelines
|
231
|
+
- See [New scheduler](https://github.com/huggingface/diffusers/issues?q=is%3Aopen+is%3Aissue+label%3A%22New+scheduler%22)
|
232
|
+
|
233
|
+
Also, say 👋 in our public Discord channel <a href="https://discord.gg/G7tWnz98XR"><img alt="Join us on Discord" src="https://img.shields.io/discord/823813159592001537?color=5865F2&logo=discord&logoColor=white"></a>. We discuss the hottest trends about diffusion models, help each other with contributions, personal projects or just hang out ☕.
|
234
|
+
|
235
|
+
|
236
|
+
## Popular Tasks & Pipelines
|
237
|
+
|
238
|
+
<table>
|
239
|
+
<tr>
|
240
|
+
<th>Task</th>
|
241
|
+
<th>Pipeline</th>
|
242
|
+
<th>🤗 Hub</th>
|
243
|
+
</tr>
|
244
|
+
<tr style="border-top: 2px solid black">
|
245
|
+
<td>Unconditional Image Generation</td>
|
246
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/ddpm"> DDPM </a></td>
|
247
|
+
<td><a href="https://huggingface.co/google/ddpm-ema-church-256"> google/ddpm-ema-church-256 </a></td>
|
248
|
+
</tr>
|
249
|
+
<tr style="border-top: 2px solid black">
|
250
|
+
<td>Text-to-Image</td>
|
251
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion/text2img">Stable Diffusion Text-to-Image</a></td>
|
252
|
+
<td><a href="https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5"> stable-diffusion-v1-5/stable-diffusion-v1-5 </a></td>
|
253
|
+
</tr>
|
254
|
+
<tr>
|
255
|
+
<td>Text-to-Image</td>
|
256
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/unclip">unCLIP</a></td>
|
257
|
+
<td><a href="https://huggingface.co/kakaobrain/karlo-v1-alpha"> kakaobrain/karlo-v1-alpha </a></td>
|
258
|
+
</tr>
|
259
|
+
<tr>
|
260
|
+
<td>Text-to-Image</td>
|
261
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/deepfloyd_if">DeepFloyd IF</a></td>
|
262
|
+
<td><a href="https://huggingface.co/DeepFloyd/IF-I-XL-v1.0"> DeepFloyd/IF-I-XL-v1.0 </a></td>
|
263
|
+
</tr>
|
264
|
+
<tr>
|
265
|
+
<td>Text-to-Image</td>
|
266
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/kandinsky">Kandinsky</a></td>
|
267
|
+
<td><a href="https://huggingface.co/kandinsky-community/kandinsky-2-2-decoder"> kandinsky-community/kandinsky-2-2-decoder </a></td>
|
268
|
+
</tr>
|
269
|
+
<tr style="border-top: 2px solid black">
|
270
|
+
<td>Text-guided Image-to-Image</td>
|
271
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/controlnet">ControlNet</a></td>
|
272
|
+
<td><a href="https://huggingface.co/lllyasviel/sd-controlnet-canny"> lllyasviel/sd-controlnet-canny </a></td>
|
273
|
+
</tr>
|
274
|
+
<tr>
|
275
|
+
<td>Text-guided Image-to-Image</td>
|
276
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/pix2pix">InstructPix2Pix</a></td>
|
277
|
+
<td><a href="https://huggingface.co/timbrooks/instruct-pix2pix"> timbrooks/instruct-pix2pix </a></td>
|
278
|
+
</tr>
|
279
|
+
<tr>
|
280
|
+
<td>Text-guided Image-to-Image</td>
|
281
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion/img2img">Stable Diffusion Image-to-Image</a></td>
|
282
|
+
<td><a href="https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5"> stable-diffusion-v1-5/stable-diffusion-v1-5 </a></td>
|
283
|
+
</tr>
|
284
|
+
<tr style="border-top: 2px solid black">
|
285
|
+
<td>Text-guided Image Inpainting</td>
|
286
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion/inpaint">Stable Diffusion Inpainting</a></td>
|
287
|
+
<td><a href="https://huggingface.co/runwayml/stable-diffusion-inpainting"> runwayml/stable-diffusion-inpainting </a></td>
|
288
|
+
</tr>
|
289
|
+
<tr style="border-top: 2px solid black">
|
290
|
+
<td>Image Variation</td>
|
291
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion/image_variation">Stable Diffusion Image Variation</a></td>
|
292
|
+
<td><a href="https://huggingface.co/lambdalabs/sd-image-variations-diffusers"> lambdalabs/sd-image-variations-diffusers </a></td>
|
293
|
+
</tr>
|
294
|
+
<tr style="border-top: 2px solid black">
|
295
|
+
<td>Super Resolution</td>
|
296
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion/upscale">Stable Diffusion Upscale</a></td>
|
297
|
+
<td><a href="https://huggingface.co/stabilityai/stable-diffusion-x4-upscaler"> stabilityai/stable-diffusion-x4-upscaler </a></td>
|
298
|
+
</tr>
|
299
|
+
<tr>
|
300
|
+
<td>Super Resolution</td>
|
301
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion/latent_upscale">Stable Diffusion Latent Upscale</a></td>
|
302
|
+
<td><a href="https://huggingface.co/stabilityai/sd-x2-latent-upscaler"> stabilityai/sd-x2-latent-upscaler </a></td>
|
303
|
+
</tr>
|
304
|
+
</table>
|
305
|
+
|
306
|
+
## Popular libraries using 🧨 Diffusers
|
307
|
+
|
308
|
+
- https://github.com/microsoft/TaskMatrix
|
309
|
+
- https://github.com/invoke-ai/InvokeAI
|
310
|
+
- https://github.com/InstantID/InstantID
|
311
|
+
- https://github.com/apple/ml-stable-diffusion
|
312
|
+
- https://github.com/Sanster/lama-cleaner
|
313
|
+
- https://github.com/IDEA-Research/Grounded-Segment-Anything
|
314
|
+
- https://github.com/ashawkey/stable-dreamfusion
|
315
|
+
- https://github.com/deep-floyd/IF
|
316
|
+
- https://github.com/bentoml/BentoML
|
317
|
+
- https://github.com/bmaltais/kohya_ss
|
318
|
+
- +14,000 other amazing GitHub repositories 💪
|
319
|
+
|
320
|
+
Thank you for using us ❤️.
|
321
|
+
|
322
|
+
## Credits
|
323
|
+
|
324
|
+
This library concretizes previous work by many different authors and would not have been possible without their great research and implementations. We'd like to thank, in particular, the following implementations which have helped us in our development and without which the API could not have been as polished today:
|
325
|
+
|
326
|
+
- @CompVis' latent diffusion models library, available [here](https://github.com/CompVis/latent-diffusion)
|
327
|
+
- @hojonathanho original DDPM implementation, available [here](https://github.com/hojonathanho/diffusion) as well as the extremely useful translation into PyTorch by @pesser, available [here](https://github.com/pesser/pytorch_diffusion)
|
328
|
+
- @ermongroup's DDIM implementation, available [here](https://github.com/ermongroup/ddim)
|
329
|
+
- @yang-song's Score-VE and Score-VP implementations, available [here](https://github.com/yang-song/score_sde_pytorch)
|
330
|
+
|
331
|
+
We also want to thank @heejkoo for the very helpful overview of papers, code and resources on diffusion models, available [here](https://github.com/heejkoo/Awesome-Diffusion-Models) as well as @crowsonkb and @rromb for useful discussions and insights.
|
332
|
+
|
333
|
+
## Citation
|
334
|
+
|
335
|
+
```bibtex
|
336
|
+
@misc{von-platen-etal-2022-diffusers,
|
337
|
+
author = {Patrick von Platen and Suraj Patil and Anton Lozhkov and Pedro Cuenca and Nathan Lambert and Kashif Rasul and Mishig Davaadorj and Dhruv Nair and Sayak Paul and William Berman and Yiyi Xu and Steven Liu and Thomas Wolf},
|
338
|
+
title = {Diffusers: State-of-the-art diffusion models},
|
339
|
+
year = {2022},
|
340
|
+
publisher = {GitHub},
|
341
|
+
journal = {GitHub repository},
|
342
|
+
howpublished = {\url{https://github.com/huggingface/diffusers}}
|
343
|
+
}
|
344
|
+
```
|
@@ -0,0 +1,239 @@
|
|
1
|
+
<!---
|
2
|
+
Copyright 2022 - The HuggingFace Team. All rights reserved.
|
3
|
+
|
4
|
+
Licensed under the Apache License, Version 2.0 (the "License");
|
5
|
+
you may not use this file except in compliance with the License.
|
6
|
+
You may obtain a copy of the License at
|
7
|
+
|
8
|
+
http://www.apache.org/licenses/LICENSE-2.0
|
9
|
+
|
10
|
+
Unless required by applicable law or agreed to in writing, software
|
11
|
+
distributed under the License is distributed on an "AS IS" BASIS,
|
12
|
+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
13
|
+
See the License for the specific language governing permissions and
|
14
|
+
limitations under the License.
|
15
|
+
-->
|
16
|
+
|
17
|
+
<p align="center">
|
18
|
+
<br>
|
19
|
+
<img src="https://raw.githubusercontent.com/huggingface/diffusers/main/docs/source/en/imgs/diffusers_library.jpg" width="400"/>
|
20
|
+
<br>
|
21
|
+
<p>
|
22
|
+
<p align="center">
|
23
|
+
<a href="https://github.com/huggingface/diffusers/blob/main/LICENSE"><img alt="GitHub" src="https://img.shields.io/github/license/huggingface/datasets.svg?color=blue"></a>
|
24
|
+
<a href="https://github.com/huggingface/diffusers/releases"><img alt="GitHub release" src="https://img.shields.io/github/release/huggingface/diffusers.svg"></a>
|
25
|
+
<a href="https://pepy.tech/project/diffusers"><img alt="GitHub release" src="https://static.pepy.tech/badge/diffusers/month"></a>
|
26
|
+
<a href="CODE_OF_CONDUCT.md"><img alt="Contributor Covenant" src="https://img.shields.io/badge/Contributor%20Covenant-2.1-4baaaa.svg"></a>
|
27
|
+
<a href="https://twitter.com/diffuserslib"><img alt="X account" src="https://img.shields.io/twitter/url/https/twitter.com/diffuserslib.svg?style=social&label=Follow%20%40diffuserslib"></a>
|
28
|
+
</p>
|
29
|
+
|
30
|
+
🤗 Diffusers is the go-to library for state-of-the-art pretrained diffusion models for generating images, audio, and even 3D structures of molecules. Whether you're looking for a simple inference solution or training your own diffusion models, 🤗 Diffusers is a modular toolbox that supports both. Our library is designed with a focus on [usability over performance](https://huggingface.co/docs/diffusers/conceptual/philosophy#usability-over-performance), [simple over easy](https://huggingface.co/docs/diffusers/conceptual/philosophy#simple-over-easy), and [customizability over abstractions](https://huggingface.co/docs/diffusers/conceptual/philosophy#tweakable-contributorfriendly-over-abstraction).
|
31
|
+
|
32
|
+
🤗 Diffusers offers three core components:
|
33
|
+
|
34
|
+
- State-of-the-art [diffusion pipelines](https://huggingface.co/docs/diffusers/api/pipelines/overview) that can be run in inference with just a few lines of code.
|
35
|
+
- Interchangeable noise [schedulers](https://huggingface.co/docs/diffusers/api/schedulers/overview) for different diffusion speeds and output quality.
|
36
|
+
- Pretrained [models](https://huggingface.co/docs/diffusers/api/models/overview) that can be used as building blocks, and combined with schedulers, for creating your own end-to-end diffusion systems.
|
37
|
+
|
38
|
+
## Installation
|
39
|
+
|
40
|
+
We recommend installing 🤗 Diffusers in a virtual environment from PyPI or Conda. For more details about installing [PyTorch](https://pytorch.org/get-started/locally/) and [Flax](https://flax.readthedocs.io/en/latest/#installation), please refer to their official documentation.
|
41
|
+
|
42
|
+
### PyTorch
|
43
|
+
|
44
|
+
With `pip` (official package):
|
45
|
+
|
46
|
+
```bash
|
47
|
+
pip install --upgrade diffusers[torch]
|
48
|
+
```
|
49
|
+
|
50
|
+
With `conda` (maintained by the community):
|
51
|
+
|
52
|
+
```sh
|
53
|
+
conda install -c conda-forge diffusers
|
54
|
+
```
|
55
|
+
|
56
|
+
### Flax
|
57
|
+
|
58
|
+
With `pip` (official package):
|
59
|
+
|
60
|
+
```bash
|
61
|
+
pip install --upgrade diffusers[flax]
|
62
|
+
```
|
63
|
+
|
64
|
+
### Apple Silicon (M1/M2) support
|
65
|
+
|
66
|
+
Please refer to the [How to use Stable Diffusion in Apple Silicon](https://huggingface.co/docs/diffusers/optimization/mps) guide.
|
67
|
+
|
68
|
+
## Quickstart
|
69
|
+
|
70
|
+
Generating outputs is super easy with 🤗 Diffusers. To generate an image from text, use the `from_pretrained` method to load any pretrained diffusion model (browse the [Hub](https://huggingface.co/models?library=diffusers&sort=downloads) for 30,000+ checkpoints):
|
71
|
+
|
72
|
+
```python
|
73
|
+
from diffusers import DiffusionPipeline
|
74
|
+
import torch
|
75
|
+
|
76
|
+
pipeline = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch.float16)
|
77
|
+
pipeline.to("cuda")
|
78
|
+
pipeline("An image of a squirrel in Picasso style").images[0]
|
79
|
+
```
|
80
|
+
|
81
|
+
You can also dig into the models and schedulers toolbox to build your own diffusion system:
|
82
|
+
|
83
|
+
```python
|
84
|
+
from diffusers import DDPMScheduler, UNet2DModel
|
85
|
+
from PIL import Image
|
86
|
+
import torch
|
87
|
+
|
88
|
+
scheduler = DDPMScheduler.from_pretrained("google/ddpm-cat-256")
|
89
|
+
model = UNet2DModel.from_pretrained("google/ddpm-cat-256").to("cuda")
|
90
|
+
scheduler.set_timesteps(50)
|
91
|
+
|
92
|
+
sample_size = model.config.sample_size
|
93
|
+
noise = torch.randn((1, 3, sample_size, sample_size), device="cuda")
|
94
|
+
input = noise
|
95
|
+
|
96
|
+
for t in scheduler.timesteps:
|
97
|
+
with torch.no_grad():
|
98
|
+
noisy_residual = model(input, t).sample
|
99
|
+
prev_noisy_sample = scheduler.step(noisy_residual, t, input).prev_sample
|
100
|
+
input = prev_noisy_sample
|
101
|
+
|
102
|
+
image = (input / 2 + 0.5).clamp(0, 1)
|
103
|
+
image = image.cpu().permute(0, 2, 3, 1).numpy()[0]
|
104
|
+
image = Image.fromarray((image * 255).round().astype("uint8"))
|
105
|
+
image
|
106
|
+
```
|
107
|
+
|
108
|
+
Check out the [Quickstart](https://huggingface.co/docs/diffusers/quicktour) to launch your diffusion journey today!
|
109
|
+
|
110
|
+
## How to navigate the documentation
|
111
|
+
|
112
|
+
| **Documentation** | **What can I learn?** |
|
113
|
+
|---------------------------------------------------------------------|-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
114
|
+
| [Tutorial](https://huggingface.co/docs/diffusers/tutorials/tutorial_overview) | A basic crash course for learning how to use the library's most important features like using models and schedulers to build your own diffusion system, and training your own diffusion model. |
|
115
|
+
| [Loading](https://huggingface.co/docs/diffusers/using-diffusers/loading) | Guides for how to load and configure all the components (pipelines, models, and schedulers) of the library, as well as how to use different schedulers. |
|
116
|
+
| [Pipelines for inference](https://huggingface.co/docs/diffusers/using-diffusers/overview_techniques) | Guides for how to use pipelines for different inference tasks, batched generation, controlling generated outputs and randomness, and how to contribute a pipeline to the library. |
|
117
|
+
| [Optimization](https://huggingface.co/docs/diffusers/optimization/fp16) | Guides for how to optimize your diffusion model to run faster and consume less memory. |
|
118
|
+
| [Training](https://huggingface.co/docs/diffusers/training/overview) | Guides for how to train a diffusion model for different tasks with different training techniques. |
|
119
|
+
## Contribution
|
120
|
+
|
121
|
+
We ❤️ contributions from the open-source community!
|
122
|
+
If you want to contribute to this library, please check out our [Contribution guide](https://github.com/huggingface/diffusers/blob/main/CONTRIBUTING.md).
|
123
|
+
You can look out for [issues](https://github.com/huggingface/diffusers/issues) you'd like to tackle to contribute to the library.
|
124
|
+
- See [Good first issues](https://github.com/huggingface/diffusers/issues?q=is%3Aopen+is%3Aissue+label%3A%22good+first+issue%22) for general opportunities to contribute
|
125
|
+
- See [New model/pipeline](https://github.com/huggingface/diffusers/issues?q=is%3Aopen+is%3Aissue+label%3A%22New+pipeline%2Fmodel%22) to contribute exciting new diffusion models / diffusion pipelines
|
126
|
+
- See [New scheduler](https://github.com/huggingface/diffusers/issues?q=is%3Aopen+is%3Aissue+label%3A%22New+scheduler%22)
|
127
|
+
|
128
|
+
Also, say 👋 in our public Discord channel <a href="https://discord.gg/G7tWnz98XR"><img alt="Join us on Discord" src="https://img.shields.io/discord/823813159592001537?color=5865F2&logo=discord&logoColor=white"></a>. We discuss the hottest trends about diffusion models, help each other with contributions, personal projects or just hang out ☕.
|
129
|
+
|
130
|
+
|
131
|
+
## Popular Tasks & Pipelines
|
132
|
+
|
133
|
+
<table>
|
134
|
+
<tr>
|
135
|
+
<th>Task</th>
|
136
|
+
<th>Pipeline</th>
|
137
|
+
<th>🤗 Hub</th>
|
138
|
+
</tr>
|
139
|
+
<tr style="border-top: 2px solid black">
|
140
|
+
<td>Unconditional Image Generation</td>
|
141
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/ddpm"> DDPM </a></td>
|
142
|
+
<td><a href="https://huggingface.co/google/ddpm-ema-church-256"> google/ddpm-ema-church-256 </a></td>
|
143
|
+
</tr>
|
144
|
+
<tr style="border-top: 2px solid black">
|
145
|
+
<td>Text-to-Image</td>
|
146
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion/text2img">Stable Diffusion Text-to-Image</a></td>
|
147
|
+
<td><a href="https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5"> stable-diffusion-v1-5/stable-diffusion-v1-5 </a></td>
|
148
|
+
</tr>
|
149
|
+
<tr>
|
150
|
+
<td>Text-to-Image</td>
|
151
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/unclip">unCLIP</a></td>
|
152
|
+
<td><a href="https://huggingface.co/kakaobrain/karlo-v1-alpha"> kakaobrain/karlo-v1-alpha </a></td>
|
153
|
+
</tr>
|
154
|
+
<tr>
|
155
|
+
<td>Text-to-Image</td>
|
156
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/deepfloyd_if">DeepFloyd IF</a></td>
|
157
|
+
<td><a href="https://huggingface.co/DeepFloyd/IF-I-XL-v1.0"> DeepFloyd/IF-I-XL-v1.0 </a></td>
|
158
|
+
</tr>
|
159
|
+
<tr>
|
160
|
+
<td>Text-to-Image</td>
|
161
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/kandinsky">Kandinsky</a></td>
|
162
|
+
<td><a href="https://huggingface.co/kandinsky-community/kandinsky-2-2-decoder"> kandinsky-community/kandinsky-2-2-decoder </a></td>
|
163
|
+
</tr>
|
164
|
+
<tr style="border-top: 2px solid black">
|
165
|
+
<td>Text-guided Image-to-Image</td>
|
166
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/controlnet">ControlNet</a></td>
|
167
|
+
<td><a href="https://huggingface.co/lllyasviel/sd-controlnet-canny"> lllyasviel/sd-controlnet-canny </a></td>
|
168
|
+
</tr>
|
169
|
+
<tr>
|
170
|
+
<td>Text-guided Image-to-Image</td>
|
171
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/pix2pix">InstructPix2Pix</a></td>
|
172
|
+
<td><a href="https://huggingface.co/timbrooks/instruct-pix2pix"> timbrooks/instruct-pix2pix </a></td>
|
173
|
+
</tr>
|
174
|
+
<tr>
|
175
|
+
<td>Text-guided Image-to-Image</td>
|
176
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion/img2img">Stable Diffusion Image-to-Image</a></td>
|
177
|
+
<td><a href="https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5"> stable-diffusion-v1-5/stable-diffusion-v1-5 </a></td>
|
178
|
+
</tr>
|
179
|
+
<tr style="border-top: 2px solid black">
|
180
|
+
<td>Text-guided Image Inpainting</td>
|
181
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion/inpaint">Stable Diffusion Inpainting</a></td>
|
182
|
+
<td><a href="https://huggingface.co/runwayml/stable-diffusion-inpainting"> runwayml/stable-diffusion-inpainting </a></td>
|
183
|
+
</tr>
|
184
|
+
<tr style="border-top: 2px solid black">
|
185
|
+
<td>Image Variation</td>
|
186
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion/image_variation">Stable Diffusion Image Variation</a></td>
|
187
|
+
<td><a href="https://huggingface.co/lambdalabs/sd-image-variations-diffusers"> lambdalabs/sd-image-variations-diffusers </a></td>
|
188
|
+
</tr>
|
189
|
+
<tr style="border-top: 2px solid black">
|
190
|
+
<td>Super Resolution</td>
|
191
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion/upscale">Stable Diffusion Upscale</a></td>
|
192
|
+
<td><a href="https://huggingface.co/stabilityai/stable-diffusion-x4-upscaler"> stabilityai/stable-diffusion-x4-upscaler </a></td>
|
193
|
+
</tr>
|
194
|
+
<tr>
|
195
|
+
<td>Super Resolution</td>
|
196
|
+
<td><a href="https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion/latent_upscale">Stable Diffusion Latent Upscale</a></td>
|
197
|
+
<td><a href="https://huggingface.co/stabilityai/sd-x2-latent-upscaler"> stabilityai/sd-x2-latent-upscaler </a></td>
|
198
|
+
</tr>
|
199
|
+
</table>
|
200
|
+
|
201
|
+
## Popular libraries using 🧨 Diffusers
|
202
|
+
|
203
|
+
- https://github.com/microsoft/TaskMatrix
|
204
|
+
- https://github.com/invoke-ai/InvokeAI
|
205
|
+
- https://github.com/InstantID/InstantID
|
206
|
+
- https://github.com/apple/ml-stable-diffusion
|
207
|
+
- https://github.com/Sanster/lama-cleaner
|
208
|
+
- https://github.com/IDEA-Research/Grounded-Segment-Anything
|
209
|
+
- https://github.com/ashawkey/stable-dreamfusion
|
210
|
+
- https://github.com/deep-floyd/IF
|
211
|
+
- https://github.com/bentoml/BentoML
|
212
|
+
- https://github.com/bmaltais/kohya_ss
|
213
|
+
- +14,000 other amazing GitHub repositories 💪
|
214
|
+
|
215
|
+
Thank you for using us ❤️.
|
216
|
+
|
217
|
+
## Credits
|
218
|
+
|
219
|
+
This library concretizes previous work by many different authors and would not have been possible without their great research and implementations. We'd like to thank, in particular, the following implementations which have helped us in our development and without which the API could not have been as polished today:
|
220
|
+
|
221
|
+
- @CompVis' latent diffusion models library, available [here](https://github.com/CompVis/latent-diffusion)
|
222
|
+
- @hojonathanho original DDPM implementation, available [here](https://github.com/hojonathanho/diffusion) as well as the extremely useful translation into PyTorch by @pesser, available [here](https://github.com/pesser/pytorch_diffusion)
|
223
|
+
- @ermongroup's DDIM implementation, available [here](https://github.com/ermongroup/ddim)
|
224
|
+
- @yang-song's Score-VE and Score-VP implementations, available [here](https://github.com/yang-song/score_sde_pytorch)
|
225
|
+
|
226
|
+
We also want to thank @heejkoo for the very helpful overview of papers, code and resources on diffusion models, available [here](https://github.com/heejkoo/Awesome-Diffusion-Models) as well as @crowsonkb and @rromb for useful discussions and insights.
|
227
|
+
|
228
|
+
## Citation
|
229
|
+
|
230
|
+
```bibtex
|
231
|
+
@misc{von-platen-etal-2022-diffusers,
|
232
|
+
author = {Patrick von Platen and Suraj Patil and Anton Lozhkov and Pedro Cuenca and Nathan Lambert and Kashif Rasul and Mishig Davaadorj and Dhruv Nair and Sayak Paul and William Berman and Yiyi Xu and Steven Liu and Thomas Wolf},
|
233
|
+
title = {Diffusers: State-of-the-art diffusion models},
|
234
|
+
year = {2022},
|
235
|
+
publisher = {GitHub},
|
236
|
+
journal = {GitHub repository},
|
237
|
+
howpublished = {\url{https://github.com/huggingface/diffusers}}
|
238
|
+
}
|
239
|
+
```
|
@@ -0,0 +1,29 @@
|
|
1
|
+
[tool.ruff]
|
2
|
+
line-length = 119
|
3
|
+
|
4
|
+
[tool.ruff.lint]
|
5
|
+
# Never enforce `E501` (line length violations).
|
6
|
+
ignore = ["C901", "E501", "E741", "F402", "F823"]
|
7
|
+
select = ["C", "E", "F", "I", "W"]
|
8
|
+
|
9
|
+
# Ignore import violations in all `__init__.py` files.
|
10
|
+
[tool.ruff.lint.per-file-ignores]
|
11
|
+
"__init__.py" = ["E402", "F401", "F403", "F811"]
|
12
|
+
"src/diffusers/utils/dummy_*.py" = ["F401"]
|
13
|
+
|
14
|
+
[tool.ruff.lint.isort]
|
15
|
+
lines-after-imports = 2
|
16
|
+
known-first-party = ["diffusers"]
|
17
|
+
|
18
|
+
[tool.ruff.format]
|
19
|
+
# Like Black, use double quotes for strings.
|
20
|
+
quote-style = "double"
|
21
|
+
|
22
|
+
# Like Black, indent with spaces, rather than tabs.
|
23
|
+
indent-style = "space"
|
24
|
+
|
25
|
+
# Like Black, respect magic trailing commas.
|
26
|
+
skip-magic-trailing-comma = false
|
27
|
+
|
28
|
+
# Like Black, automatically detect the appropriate line ending.
|
29
|
+
line-ending = "auto"
|