diffusers
b6e0b016 - Lazy Import for Diffusers (#4829)

Commit
1 year ago
Lazy Import for Diffusers (#4829) * initial commit * move modules to import struct * add dummy objects and _LazyModule * add lazy import to schedulers * clean up unused imports * lazy import on models module * lazy import for schedulers module * add lazy import to pipelines module * lazy import altdiffusion * lazy import audio diffusion * lazy import audioldm * lazy import consistency model * lazy import controlnet * lazy import dance diffusion ddim ddpm * lazy import deepfloyd * lazy import kandinksy * lazy imports * lazy import semantic diffusion * lazy imports * lazy import stable diffusion * move sd output to its own module * clean up * lazy import t2iadapter * lazy import unclip * lazy import versatile and vq diffsuion * lazy import vq diffusion * helper to fetch objects from modules * lazy import sdxl * lazy import txt2vid * lazy import stochastic karras * fix model imports * fix bug * lazy import * clean up * clean up * fixes for tests * fixes for tests * clean up * remove import of torch_utils from utils module * clean up * clean up * fix mistake import statement * dedicated modules for exporting and loading * remove testing utils from utils module * fixes from merge conflicts * Update src/diffusers/pipelines/kandinsky2_2/__init__.py * fix docs * fix alt diffusion copied from * fix check dummies * fix more docs * remove accelerate import from utils module * add type checking * make style * fix check dummies * remove torch import from xformers check * clean up error message * fixes after upstream merges * dummy objects fix * fix tests * remove unused module import --------- Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Author
DN6 DN6
Parents
  • docs/source/en
    • api
      • File
        utilities.md
    • using-diffusers
      • File
        reproducibility.md
  • examples/community
    • File
      clip_guided_images_mixing_stable_diffusion.py
    • File
      clip_guided_stable_diffusion_img2img.py
    • File
      ddim_noise_comparative_analysis.py
    • File
      lpw_stable_diffusion.py
    • File
      lpw_stable_diffusion_xl.py
    • File
      pipeline_fabric.py
    • File
      pipeline_zero1to3.py
    • File
      run_onnx_controlnet.py
    • File
      run_tensorrt_controlnet.py
    • File
      stable_diffusion_controlnet_img2img.py
    • File
      stable_diffusion_controlnet_inpaint.py
    • File
      stable_diffusion_controlnet_inpaint_img2img.py
    • File
      stable_diffusion_controlnet_reference.py
    • File
      stable_diffusion_ipex.py
    • File
      stable_diffusion_reference.py
    • File
      stable_diffusion_repaint.py
    • File
      stable_diffusion_xl_reference.py
    • File
      stable_unclip.py
    • File
      unclip_image_interpolation.py
    • File
      unclip_text_interpolation.py
  • src/diffusers
    • File
      __init__.py
    • experimental/rl
      • File
        value_guided_sampling.py
    • models
      • File
        __init__.py
      • File
        attention.py
      • File
        attention_processor.py
      • File
        autoencoder_asym_kl.py
      • File
        autoencoder_kl.py
      • File
        autoencoder_tiny.py
      • File
        vae.py
      • File
        vq_model.py
    • pipelines
      • File
        __init__.py
      • alt_diffusion
        • File
          __init__.py
        • File
          pipeline_alt_diffusion.py
        • File
          pipeline_alt_diffusion_img2img.py
        • File
          pipeline_output.py
      • audio_diffusion
        • File
          __init__.py
        • File
          pipeline_audio_diffusion.py
      • audioldm
        • File
          __init__.py
        • File
          pipeline_audioldm.py
      • audioldm2
        • File
          __init__.py
        • File
          pipeline_audioldm2.py
      • consistency_models
        • File
          __init__.py
        • File
          pipeline_consistency_models.py
      • controlnet
        • File
          __init__.py
        • File
          pipeline_controlnet.py
        • File
          pipeline_controlnet_img2img.py
        • File
          pipeline_controlnet_inpaint.py
        • File
          pipeline_controlnet_inpaint_sd_xl.py
        • File
          pipeline_controlnet_sd_xl.py
        • File
          pipeline_controlnet_sd_xl_img2img.py
      • dance_diffusion
        • File
          __init__.py
        • File
          pipeline_dance_diffusion.py
      • ddim
        • File
          __init__.py
        • File
          pipeline_ddim.py
      • ddpm
        • File
          __init__.py
        • File
          pipeline_ddpm.py
      • deepfloyd_if
        • File
          __init__.py
        • File
          pipeline_if.py
        • File
          pipeline_if_img2img.py
        • File
          pipeline_if_img2img_superresolution.py
        • File
          pipeline_if_inpainting.py
        • File
          pipeline_if_inpainting_superresolution.py
        • File
          pipeline_if_superresolution.py
        • File
          pipeline_output.py
      • dit
        • File
          __init__.py
        • File
          pipeline_dit.py
      • kandinsky
        • File
          __init__.py
        • File
          pipeline_kandinsky.py
        • File
          pipeline_kandinsky_img2img.py
        • File
          pipeline_kandinsky_inpaint.py
        • File
          pipeline_kandinsky_prior.py
      • kandinsky2_2
        • File
          __init__.py
        • File
          pipeline_kandinsky2_2.py
        • File
          pipeline_kandinsky2_2_controlnet.py
        • File
          pipeline_kandinsky2_2_controlnet_img2img.py
        • File
          pipeline_kandinsky2_2_img2img.py
        • File
          pipeline_kandinsky2_2_inpainting.py
        • File
          pipeline_kandinsky2_2_prior.py
        • File
          pipeline_kandinsky2_2_prior_emb2emb.py
      • latent_diffusion
        • File
          __init__.py
        • File
          pipeline_latent_diffusion.py
        • File
          pipeline_latent_diffusion_superresolution.py
      • latent_diffusion_uncond
        • File
          __init__.py
        • File
          pipeline_latent_diffusion_uncond.py
      • musicldm
        • File
          __init__.py
        • File
          pipeline_musicldm.py
      • paint_by_example
        • File
          __init__.py
        • File
          pipeline_paint_by_example.py
      • File
        pipeline_utils.py
      • pndm
        • File
          __init__.py
        • File
          pipeline_pndm.py
      • repaint
        • File
          __init__.py
        • File
          pipeline_repaint.py
      • score_sde_ve
        • File
          __init__.py
        • File
          pipeline_score_sde_ve.py
      • semantic_stable_diffusion
        • File
          __init__.py
        • File
          pipeline_output.py
        • File
          pipeline_semantic_stable_diffusion.py
      • shap_e
        • File
          __init__.py
        • File
          pipeline_shap_e.py
        • File
          pipeline_shap_e_img2img.py
      • spectrogram_diffusion
        • File
          __init__.py
        • File
          pipeline_spectrogram_diffusion.py
      • stable_diffusion
        • File
          __init__.py
        • File
          pipeline_cycle_diffusion.py
        • File
          pipeline_output.py
        • File
          pipeline_stable_diffusion.py
        • File
          pipeline_stable_diffusion_attend_and_excite.py
        • File
          pipeline_stable_diffusion_depth2img.py
        • File
          pipeline_stable_diffusion_diffedit.py
        • File
          pipeline_stable_diffusion_gligen.py
        • File
          pipeline_stable_diffusion_gligen_text_image.py
        • File
          pipeline_stable_diffusion_image_variation.py
        • File
          pipeline_stable_diffusion_img2img.py
        • File
          pipeline_stable_diffusion_inpaint.py
        • File
          pipeline_stable_diffusion_inpaint_legacy.py
        • File
          pipeline_stable_diffusion_instruct_pix2pix.py
        • File
          pipeline_stable_diffusion_k_diffusion.py
        • File
          pipeline_stable_diffusion_latent_upscale.py
        • File
          pipeline_stable_diffusion_ldm3d.py
        • File
          pipeline_stable_diffusion_model_editing.py
        • File
          pipeline_stable_diffusion_panorama.py
        • File
          pipeline_stable_diffusion_paradigms.py
        • File
          pipeline_stable_diffusion_pix2pix_zero.py
        • File
          pipeline_stable_diffusion_sag.py
        • File
          pipeline_stable_diffusion_upscale.py
        • File
          pipeline_stable_unclip.py
        • File
          pipeline_stable_unclip_img2img.py
      • stable_diffusion_safe
        • File
          pipeline_stable_diffusion_safe.py
      • stable_diffusion_xl
        • File
          __init__.py
        • File
          pipeline_output.py
        • File
          pipeline_stable_diffusion_xl.py
        • File
          pipeline_stable_diffusion_xl_img2img.py
        • File
          pipeline_stable_diffusion_xl_inpaint.py
        • File
          pipeline_stable_diffusion_xl_instruct_pix2pix.py
      • stochastic_karras_ve
        • File
          __init__.py
        • File
          pipeline_stochastic_karras_ve.py
      • t2i_adapter
        • File
          __init__.py
        • File
          pipeline_stable_diffusion_adapter.py
        • File
          pipeline_stable_diffusion_xl_adapter.py
      • text_to_video_synthesis
        • File
          __init__.py
        • File
          pipeline_output.py
        • File
          pipeline_text_to_video_synth.py
        • File
          pipeline_text_to_video_synth_img2img.py
      • unclip
        • File
          __init__.py
        • File
          pipeline_unclip.py
        • File
          pipeline_unclip_image_variation.py
      • unidiffuser
        • File
          __init__.py
        • File
          pipeline_unidiffuser.py
      • versatile_diffusion
        • File
          __init__.py
        • File
          pipeline_versatile_diffusion_dual_guided.py
        • File
          pipeline_versatile_diffusion_image_variation.py
        • File
          pipeline_versatile_diffusion_text_to_image.py
      • vq_diffusion
        • File
          __init__.py
      • wuerstchen
        • File
          __init__.py
        • File
          modeling_paella_vq_model.py
        • File
          pipeline_wuerstchen.py
        • File
          pipeline_wuerstchen_prior.py
    • schedulers
      • File
        __init__.py
      • File
        scheduling_consistency_models.py
      • File
        scheduling_ddim.py
      • File
        scheduling_ddim_parallel.py
      • File
        scheduling_ddpm.py
      • File
        scheduling_ddpm_parallel.py
      • File
        scheduling_ddpm_wuerstchen.py
      • File
        scheduling_dpmsolver_multistep.py
      • File
        scheduling_dpmsolver_multistep_inverse.py
      • File
        scheduling_euler_ancestral_discrete.py
      • File
        scheduling_euler_discrete.py
      • File
        scheduling_k_dpm_2_ancestral_discrete.py
      • File
        scheduling_karras_ve.py
      • File
        scheduling_repaint.py
      • File
        scheduling_sde_ve.py
      • File
        scheduling_sde_vp.py
      • File
        scheduling_unclip.py
    • utils
      • File
        __init__.py
      • File
        export_utils.py
      • File
        import_utils.py
      • File
        loading_utils.py
  • tests
    • models
      • File
        test_layers_utils.py
      • File
        test_lora_layers.py
      • File
        test_modeling_common.py
      • File
        test_models_prior.py
      • File
        test_models_unet_1d.py
      • File
        test_models_unet_2d.py
      • File
        test_models_unet_2d_condition.py
      • File
        test_models_unet_3d_condition.py
      • File
        test_models_vae.py
      • File
        test_models_vq.py
      • File
        test_unet_2d_blocks.py
      • File
        test_unet_blocks_common.py
    • pipelines
      • altdiffusion
        • File
          test_alt_diffusion.py
        • File
          test_alt_diffusion_img2img.py
      • audio_diffusion
        • File
          test_audio_diffusion.py
      • audioldm
        • File
          test_audioldm.py
      • audioldm2
        • File
          test_audioldm2.py
      • consistency_models
        • File
          test_consistency_models.py
      • controlnet
        • File
          test_controlnet.py
        • File
          test_controlnet_img2img.py
        • File
          test_controlnet_inpaint.py
        • File
          test_controlnet_inpaint_sdxl.py
        • File
          test_controlnet_sdxl.py
        • File
          test_controlnet_sdxl_img2img.py
        • File
          test_flax_controlnet.py
      • dance_diffusion
        • File
          test_dance_diffusion.py
      • deepfloyd_if
        • File
          test_if_img2img.py
        • File
          test_if_img2img_superresolution.py
        • File
          test_if_inpainting.py
        • File
          test_if_inpainting_superresolution.py
        • File
          test_if_superresolution.py
      • dit
        • File
          test_dit.py
      • kandinsky
        • File
          test_kandinsky.py
        • File
          test_kandinsky_combined.py
        • File
          test_kandinsky_img2img.py
        • File
          test_kandinsky_inpaint.py
        • File
          test_kandinsky_prior.py
      • kandinsky_v22
        • File
          test_kandinsky.py
        • File
          test_kandinsky_combined.py
        • File
          test_kandinsky_controlnet.py
        • File
          test_kandinsky_controlnet_img2img.py
        • File
          test_kandinsky_img2img.py
        • File
          test_kandinsky_inpaint.py
        • File
          test_kandinsky_prior.py
        • File
          test_kandinsky_prior_emb2emb.py
      • latent_diffusion
        • File
          test_latent_diffusion_superresolution.py
      • musicldm
        • File
          test_musicldm.py
      • paint_by_example
        • File
          test_paint_by_example.py
      • semantic_stable_diffusion
        • File
          test_semantic_diffusion.py
      • shap_e
        • File
          test_shap_e.py
        • File
          test_shap_e_img2img.py
      • spectrogram_diffusion
        • File
          test_spectrogram_diffusion.py
      • stable_diffusion
        • File
          test_cycle_diffusion.py
        • File
          test_onnx_stable_diffusion_img2img.py
        • File
          test_onnx_stable_diffusion_upscale.py
        • File
          test_stable_diffusion.py
        • File
          test_stable_diffusion_adapter.py
        • File
          test_stable_diffusion_gligen_text_image.py
        • File
          test_stable_diffusion_image_variation.py
        • File
          test_stable_diffusion_img2img.py
        • File
          test_stable_diffusion_inpaint.py
        • File
          test_stable_diffusion_inpaint_legacy.py
        • File
          test_stable_diffusion_instruction_pix2pix.py
        • File
          test_stable_diffusion_k_diffusion.py
        • File
          test_stable_diffusion_ldm3d.py
        • File
          test_stable_diffusion_model_editing.py
        • File
          test_stable_diffusion_panorama.py
        • File
          test_stable_diffusion_paradigms.py
        • File
          test_stable_diffusion_pix2pix_zero.py
        • File
          test_stable_diffusion_sag.py
      • stable_diffusion_2
        • File
          test_stable_diffusion.py
        • File
          test_stable_diffusion_attend_and_excite.py
        • File
          test_stable_diffusion_depth.py
        • File
          test_stable_diffusion_diffedit.py
        • File
          test_stable_diffusion_flax.py
        • File
          test_stable_diffusion_flax_inpaint.py
        • File
          test_stable_diffusion_inpaint.py
        • File
          test_stable_diffusion_latent_upscale.py
        • File
          test_stable_diffusion_upscale.py
        • File
          test_stable_diffusion_v_pred.py
      • stable_diffusion_safe
        • File
          test_safe_diffusion.py
      • stable_diffusion_xl
        • File
          test_stable_diffusion_xl.py
        • File
          test_stable_diffusion_xl_adapter.py
        • File
          test_stable_diffusion_xl_img2img.py
        • File
          test_stable_diffusion_xl_inpaint.py
        • File
          test_stable_diffusion_xl_instruction_pix2pix.py
      • File
        test_pipelines.py
      • File
        test_pipelines_auto.py
      • File
        test_pipelines_common.py
      • text_to_video
        • File
          test_text_to_video.py
        • File
          test_text_to_video_zero.py
        • File
          test_video_to_video.py
      • unclip
        • File
          test_unclip.py
        • File
          test_unclip_image_variation.py
      • unidiffuser
        • File
          test_unidiffuser.py
      • vq_diffusion
        • File
          test_vq_diffusion.py
      • wuerstchen
        • File
          test_wuerstchen_combined.py
        • File
          test_wuerstchen_decoder.py
        • File
          test_wuerstchen_prior.py
    • schedulers
      • File
        test_scheduler_dpm_sde.py
      • File
        test_scheduler_euler.py
      • File
        test_scheduler_euler_ancestral.py
      • File
        test_scheduler_heun.py
      • File
        test_scheduler_kdpm2_ancestral.py
      • File
        test_scheduler_kdpm2_discrete.py
      • File
        test_scheduler_lms.py
      • File
        test_schedulers.py
  • utils
    • File
      check_copies.py
    • File
      check_dummies.py