• clarencechen's avatar
    Diffedit Zero-Shot Inpainting Pipeline (#2837) · be0bfcec
    clarencechen authored
    * Update Pix2PixZero Auto-correlation Loss
    
    * Add Stable Diffusion DiffEdit pipeline
    
    * Add draft documentation and import code
    
    * Bugfixes and refactoring
    
    * Add option to not decode latents in the inversion process
    
    * Harmonize preprocessing
    
    * Revert "Update Pix2PixZero Auto-correlation Loss"
    
    This reverts commit b218062fed08d6cc164206d6cb852b2b7b00847a.
    
    * Update annotations
    
    * rename `compute_mask` to `generate_mask`
    
    * Update documentation
    
    * Update docs
    
    * Update Docs
    
    * Fix copy
    
    * Change shape of output latents to batch first
    
    * Update docs
    
    * Add first draft for tests
    
    * Bugfix and update tests
    
    * Add `cross_attention_kwargs` support for all pipeline methods
    
    * Fix Copies
    
    * Add support for PIL image latents
    
    Add support for mask broadcasting
    
    Update docs and tests
    
    Align `mask` argument to `mask_image`
    
    Remove height and width arguments
    
    * Enable MPS Tests
    
    * Move example docstrings
    
    * Fix test
    
    * Fix test
    
    * fix pipeline inheritance
    
    * Harmonize `prepare_image_latents` with StableDiffusionPix2PixZeroPipeline
    
    * Register modules set to `None` in config for `test_save_load_optional_components`
    
    * Move fixed logic to specific test class
    
    * Clean changes to other pipelines
    
    * Update new tests to coordinate with #2953
    
    * Update slow tests for better results
    
    * Safety to avoid potential problems with torch.inference_mode
    
    * Add reference in SD Pipeline Overview
    
    * Fix tests again
    
    * Enforce determinism in noise for generate_mask
    
    * Fix copies
    
    * Widen test tolerance for fp16 based on `test_stable_diffusion_upscale_pipeline_fp16`
    
    * Add LoraLoaderMixin and update `prepare_image_latents`
    
    * clean up repeat and reg
    
    * bugfix
    
    * Remove invalid args from docs
    
    Suppress spurious warning by repeating image before latent to mask gen
    be0bfcec
__init__.py 5.47 KB