• Suraj Patil's avatar
    StableDiffusionDepth2ImgPipeline (#1531) · 5383188c
    Suraj Patil authored
    
    
    * begin depth pipeline
    
    * add depth estimation model
    
    * fix prepare_depth_mask
    
    * add a comment about autocast
    
    * copied from, quality, cleanup
    
    * begin tests
    
    * handle tensors
    
    * norm image tensor
    
    * fix batch size
    
    * fix tests
    
    * fix enable_sequential_cpu_offload
    
    * fix save load
    
    * fix test_save_load_float16
    
    * fix test_save_load_optional_components
    
    * fix test_float16_inference
    
    * fix test_cpu_offload_forward_pass
    
    * fix test_dict_tuple_outputs_equivalent
    
    * up
    
    * fix fast tests
    
    * fix test_stable_diffusion_img2img_multiple_init_images
    
    * fix few more fast tests
    
    * don't use device map for DPT
    
    * fix test_stable_diffusion_pipeline_with_sequential_cpu_offloading
    
    * accept external depth maps
    
    * prepare_depth_mask -> prepare_depth_map
    
    * fix file name
    
    * fix file name
    
    * quality
    
    * check transformers version
    
    * fix test names
    
    * use skipif
    
    * fix import
    
    * add docs
    
    * skip tests on mps
    
    * correct version
    
    * uP
    
    * Update docs/source/api/pipelines/stable_diffusion_2.mdx
    
    * fix fix-copies
    
    * fix fix-copies
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    Co-authored-by: default avataranton- <anton@huggingface.co>
    5383188c
dependency_versions_table.py 1.12 KB